LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.cfg
setup.py
bagua/__init__.py
bagua/bagua_define.py
bagua/version.py
bagua.egg-info/PKG-INFO
bagua.egg-info/SOURCES.txt
bagua.egg-info/dependency_links.txt
bagua.egg-info/entry_points.txt
bagua.egg-info/not-zip-safe
bagua.egg-info/requires.txt
bagua.egg-info/top_level.txt
bagua/distributed/__init__.py
bagua/distributed/launch.py
bagua/distributed/run.py
bagua/script/__init__.py
bagua/script/bagua_sys_perf
bagua/script/baguarun.py
bagua/service/__init__.py
bagua/service/autotune_service.py
bagua/service/autotune_system.py
bagua/service/autotune_task_manager.py
bagua/service/bayesian_optimizer.py
bagua/torch_api/__init__.py
bagua/torch_api/bucket.py
bagua/torch_api/communication.py
bagua/torch_api/distributed.py
bagua/torch_api/env.py
bagua/torch_api/tensor.py
bagua/torch_api/utils.py
bagua/torch_api/algorithms/__init__.py
bagua/torch_api/algorithms/async_model_average.py
bagua/torch_api/algorithms/base.py
bagua/torch_api/algorithms/bytegrad.py
bagua/torch_api/algorithms/decentralized.py
bagua/torch_api/algorithms/gradient_allreduce.py
bagua/torch_api/algorithms/q_adam.py
bagua/torch_api/checkpoint/__init__.py
bagua/torch_api/checkpoint/checkpointing.py
bagua/torch_api/contrib/__init__.py
bagua/torch_api/contrib/cache_loader.py
bagua/torch_api/contrib/cached_dataset.py
bagua/torch_api/contrib/load_balancing_data_loader.py
bagua/torch_api/contrib/sync_batchnorm.py
bagua/torch_api/contrib/fuse/__init__.py
bagua/torch_api/contrib/fuse/optimizer.py
bagua/torch_api/contrib/utils/__init__.py
bagua/torch_api/contrib/utils/redis_store.py
bagua/torch_api/contrib/utils/store.py
bagua/torch_api/data_parallel/__init__.py
bagua/torch_api/data_parallel/bagua_distributed.py
bagua/torch_api/data_parallel/distributed.py
bagua/torch_api/data_parallel/functional.py
bagua/torch_api/model_parallel/__init__.py
bagua/torch_api/model_parallel/moe/__init__.py
bagua/torch_api/model_parallel/moe/experts.py
bagua/torch_api/model_parallel/moe/layer.py
bagua/torch_api/model_parallel/moe/sharded_moe.py
bagua/torch_api/model_parallel/moe/utils.py
bagua_core/__init__.py
bagua_core/_environment.py
bagua_core/bagua_install_deps.py
rust/bagua-core/Cargo.toml
rust/bagua-core/bagua-core-c/Cargo.toml
rust/bagua-core/bagua-core-c/src/lib.rs
rust/bagua-core/bagua-core-internal/Cargo.toml
rust/bagua-core/bagua-core-internal/build.rs
rust/bagua-core/bagua-core-internal/cpp/include/bagua_utils.h
rust/bagua-core/bagua-core-internal/cpp/include/dbg.h
rust/bagua-core/bagua-core-internal/kernels/bagua_kernels.cu
rust/bagua-core/bagua-core-internal/src/cuda_utils.rs
rust/bagua-core/bagua-core-internal/src/events.rs
rust/bagua-core/bagua-core-internal/src/lib.rs
rust/bagua-core/bagua-core-internal/src/torch_ffi.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/centralized_full_precision_synchronous.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/centralized_low_precision_synchronous.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/decentralized_full_precision_asynchronous.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/decentralized_full_precision_synchronous.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/decentralized_low_precision_synchronous.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/mod.rs
rust/bagua-core/bagua-core-internal/src/comm_ops/python_ffi_op.rs
rust/bagua-core/bagua-core-internal/src/communicators/mod.rs
rust/bagua-core/bagua-core-internal/src/datatypes/mod.rs
rust/bagua-core/bagua-core-internal/src/kernels/mod.rs
rust/bagua-core/bagua-core-internal/src/resource_pool/mod.rs
rust/bagua-core/bagua-core-internal/third_party/Aluminum/.gitignore
rust/bagua-core/bagua-core-internal/third_party/Aluminum/.gitmodules
rust/bagua-core/bagua-core-internal/third_party/Aluminum/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/CONTRIBUTORS
rust/bagua-core/bagua-core-internal/third_party/Aluminum/LICENSE
rust/bagua-core/bagua-core-internal/third_party/Aluminum/README.md
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_events.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_ops.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_utils.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_utils_ht.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_utils_mpi.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_utils_nccl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/benchmark_waits.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/plot_benchmarks.py
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/run_benchmarks.py
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/wait.cu
rust/bagua-core/bagua-core-internal/third_party/Aluminum/benchmark/wait.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/Al_config.hpp.in
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/AluminumConfig.cmake.in
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/FindCUB.cmake
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/FindHWLOC.cmake
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/FindNCCL.cmake
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/FindNVTX.cmake
rust/bagua-core/bagua-core-internal/third_party/Aluminum/cmake/HipBuildSystem.cmake
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/Al.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/base.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht_impl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/internal.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mempool.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_comm_and_stream_wrapper.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda_impl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_impl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/nccl_impl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/profiling.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/progress.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/state.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/trace.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/tuning_params.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/cuda.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/cuda_mempool.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/events.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/gpu_status_flag.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/gpu_wait.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/helper_kernels.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/streams.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/cuda/sync_memory.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/allgather.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/allgatherv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/allreduce.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/alltoall.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/alltoallv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/barrier.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/base_state.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/bcast.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/communicator.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/gather.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/gatherv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/pt2pt.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/reduce.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/reduce_scatter.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/reduce_scatterv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/scatter.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/ht/scatterv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/allgather.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/allgatherv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/allreduce.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/alltoall.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/alltoallv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/barrier.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/base_state.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/bcast.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/communicator.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/gather.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/gatherv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/pt2pt.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/reduce.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/reduce_scatter.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/reduce_scatterv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/scatter.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/scatterv.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi/utils.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/communicator.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/rma.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/rma_ipc.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/rma_null.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/rma_self.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/mpi_cuda/util.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/caching_allocator.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/locked_resource_pool.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/meta.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/mpsc_queue.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/spsc_queue.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/include/aluminum/utils/utils.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/Al.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/ht_impl.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/mempool.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/mpi_cuda_impl.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/mpi_impl.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/nccl_impl.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/profiling.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/progress.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/trace.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/cuda/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/cuda/cuda.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/cuda/gpu_status_flag.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/cuda/gpu_wait.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/cuda/helper_kernels.cu
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/cuda/streams.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/mpi_cuda/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/mpi_cuda/communicator.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/src/mpi_cuda/rma.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/cuda_vector.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/hang_watchdog.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/op_dispatcher.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/op_runner_impl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/run_tests.py
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_exchange.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_ops.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_rma_halo_exchange.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_rma_ring.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_stream_mem_ops.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_utils.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_utils_ht.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_utils_mpi.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_utils_mpi_cuda.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/test/test_utils_nccl.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/.git
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/.gitignore
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/.travis.yml
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/CHANGELOG.md
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/INSTALL
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/LICENSE
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/README.md
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/cxxopts-config.cmake.in
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/include/cxxopts.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/src/.gitignore
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/src/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/src/example.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/.gitignore
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/catch.hpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/link_a.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/link_b.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/main.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/options.cpp
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/add-subdirectory-test/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/Aluminum/third_party/cxxopts/test/find-package-test/CMakeLists.txt
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.cproject
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.project
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/CHANGE_LOG.TXT
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/LICENSE.TXT
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/README.md
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/common.mk
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/eclipse code style profile.xml
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.settings/.gitignore
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.settings/org.eclipse.cdt.codan.core.prefs
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.settings/org.eclipse.cdt.core.prefs
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.settings/org.eclipse.cdt.ui.prefs
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.settings/org.eclipse.core.runtime.prefs
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/cub.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_allocator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_arch.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_debug.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_device.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_macro.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_namespace.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_ptx.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/util_type.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_histogram.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_radix_sort_downsweep.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_radix_sort_upsweep.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_reduce_by_key.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_rle.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_scan.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_segment_fixup.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_select_if.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/agent_spmv_orig.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/agent/single_pass_scan_operators.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_adjacent_difference.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_discontinuity.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_exchange.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_histogram.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_load.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_radix_rank.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_radix_sort.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_raking_layout.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_scan.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_shuffle.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/block_store.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_histogram_atomic.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_histogram_sort.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_reduce_raking.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_reduce_raking_commutative_only.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_reduce_warp_reductions.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_scan_raking.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_scan_warp_scans.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_scan_warp_scans2.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/block/specializations/block_scan_warp_scans3.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_histogram.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_partition.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_radix_sort.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_run_length_encode.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_scan.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_segmented_radix_sort.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_segmented_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_select.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/device_spmv.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_histogram.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_radix_sort.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_reduce_by_key.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_rle.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_scan.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_select_if.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/device/dispatch/dispatch_spmv_orig.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/grid/grid_barrier.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/grid/grid_even_share.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/grid/grid_mapping.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/grid/grid_queue.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/host/mutex.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/arg_index_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/cache_modified_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/cache_modified_output_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/constant_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/counting_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/discard_output_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/tex_obj_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/tex_ref_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/iterator/transform_input_iterator.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/thread/thread_load.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/thread/thread_operators.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/thread/thread_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/thread/thread_scan.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/thread/thread_search.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/thread/thread_store.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/warp/warp_reduce.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/warp/warp_scan.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/warp/specializations/warp_reduce_shfl.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/warp/specializations/warp_reduce_smem.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/warp/specializations/warp_scan_shfl.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/cub/warp/specializations/warp_scan_smem.cuh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/block/.gitignore
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/block/Makefile
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/block/example_block_radix_sort.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/block/example_block_reduce.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/block/example_block_scan.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/block/reduce_by_key.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/.gitignore
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/Makefile
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_partition_flagged.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_partition_if.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_radix_sort.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_reduce.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_scan.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_select_flagged.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_select_if.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_select_unique.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/examples/device/example_device_sort_find_non_trivial_runs.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/.gitignore
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/Makefile
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/histogram_compare.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/sparse_matrix.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/spmv_compare.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/spmv_script.sh
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/defunct/example_coo_spmv.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/defunct/test_device_seg_reduce.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/histogram/histogram_cub.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/histogram/histogram_gmem_atomics.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/histogram/histogram_smem_atomics.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/.gitignore
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/Makefile
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/half.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/link_a.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/link_b.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/link_main.cpp
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/mersenne.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_allocator.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_block_histogram.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_block_load_store.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_block_radix_sort.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_block_reduce.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_block_scan.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_histogram.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_radix_sort.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_reduce.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_reduce_by_key.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_run_length_encode.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_scan.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_select_if.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_device_select_unique.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_grid_barrier.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_iterator.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_util.h
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_warp_reduce.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/test_warp_scan.cu
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/tune/.gitignore
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/tune/Makefile
rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/tune/tune_device_reduce.cu
rust/bagua-core/bagua-core-py/Cargo.toml
rust/bagua-core/bagua-core-py/src/lib.rs
tests/__init__.py
tests/test_bagua_define.py
tests/internal/__init__.py
tests/internal/common_utils.py
tests/internal/compressor.py
tests/internal/multi_process.py
tests/internal/torch/__init__.py
tests/internal/torch/_core.py
tests/internal/torch/common_cuda.py
tests/internal/torch/common_distributed.py
tests/internal/torch/common_utils.py
tests/internal/torch/expecttest.py
tests/pytorch_lightning/__init__.py
tests/pytorch_lightning/boring_model.py
tests/pytorch_lightning/test_bagua_strategy.py