├── .buildkite ├── pipeline.yml └── scripts │ ├── benchmark.sh │ ├── benchmark_master.sh │ ├── benchmark_worker.sh │ ├── example.sh │ ├── install_bagua.sh │ └── run_pytest.sh ├── .github ├── CODEOWNERS ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── dependabot.yml ├── issue-close-app.yml └── workflows │ ├── bagua-net-ci.yml │ ├── bagua-pypi-publish.yml │ ├── bagua-python-package-check.yml │ ├── black.yml │ ├── dockerhub.yml │ ├── flake8.yml │ ├── pytype.yml │ └── rustfmt.yml ├── .gitignore ├── .gitmodules ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── bagua ├── __init__.py ├── bagua_define.py ├── distributed │ ├── __init__.py │ ├── launch.py │ └── run.py ├── script │ ├── __init__.py │ ├── bagua_sys_perf │ └── baguarun.py ├── service │ ├── __init__.py │ ├── autotune_service.py │ ├── autotune_system.py │ ├── autotune_task_manager.py │ └── bayesian_optimizer.py ├── torch_api │ ├── __init__.py │ ├── algorithms │ │ ├── __init__.py │ │ ├── async_model_average.py │ │ ├── base.py │ │ ├── bytegrad.py │ │ ├── decentralized.py │ │ ├── gradient_allreduce.py │ │ └── q_adam.py │ ├── bucket.py │ ├── checkpoint │ │ ├── __init__.py │ │ └── checkpointing.py │ ├── communication.py │ ├── contrib │ │ ├── __init__.py │ │ ├── cache_loader.py │ │ ├── cached_dataset.py │ │ ├── fuse │ │ │ ├── __init__.py │ │ │ └── optimizer.py │ │ ├── load_balancing_data_loader.py │ │ ├── sync_batchnorm.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── redis_store.py │ │ │ └── store.py │ ├── data_parallel │ │ ├── __init__.py │ │ ├── bagua_distributed.py │ │ ├── distributed.py │ │ └── functional.py │ ├── distributed.py │ ├── env.py │ ├── model_parallel │ │ ├── __init__.py │ │ └── moe │ │ │ ├── __init__.py │ │ │ ├── experts.py │ │ │ ├── layer.py │ │ │ ├── sharded_moe.py │ │ │ └── utils.py │ ├── tensor.py │ └── utils.py └── version.py ├── bagua_core ├── __init__.py ├── _environment.py └── bagua_install_deps.py ├── cliff.toml ├── docker ├── Dockerfile.manylinux-cuda102 ├── Dockerfile.manylinux-cuda111 ├── Dockerfile.manylinux-cuda113 ├── Dockerfile.manylinux-cuda115 ├── Dockerfile.manylinux-cuda116 ├── Dockerfile.manylinux-cuda117 ├── Dockerfile.pytorch-1.10.0-cuda11.3-cudnn8 ├── Dockerfile.pytorch-1.11.0-cuda11.3-cudnn8 ├── Dockerfile.pytorch-1.12.0-cuda11.3-cudnn8 ├── Dockerfile.pytorch-1.13.0-cuda11.6-cudnn8 ├── Dockerfile.pytorch-1.9.0-cuda10.2-cudnn7 ├── Dockerfile.pytorch-1.9.0-cuda11.1-cudnn8 └── Dockerfile.pytorch-2.0.0-cuda11.7-cudnn8 ├── docs ├── _autoapi_templates │ ├── index.rst │ └── python │ │ ├── attribute.rst │ │ ├── class.rst │ │ ├── data.rst │ │ ├── exception.rst │ │ ├── function.rst │ │ ├── method.rst │ │ ├── module.rst │ │ └── package.rst ├── conf.py ├── doc-requirements.txt └── index.rst ├── examples ├── LICENSE ├── README.md ├── benchmark │ ├── README.md │ ├── requirements.txt │ └── synthetic_benchmark.py ├── communication_primitives │ ├── README.md │ ├── main.py │ └── requirements.txt ├── elastic_training │ ├── README.md │ ├── main.py │ └── requirements.txt ├── imagenet │ ├── README.md │ ├── main.py │ └── requirements.txt ├── mnist │ ├── README.md │ ├── main.py │ └── requirements.txt ├── moe │ ├── README.md │ ├── mnist_main.py │ └── requirements.txt └── squad │ ├── README.md │ ├── main.py │ └── requirements.txt ├── figures └── logo.png ├── install.sh ├── install_master.sh ├── pyproject.toml ├── rust ├── bagua-core │ ├── .gitignore │ ├── Cargo.lock │ ├── Cargo.toml │ ├── README.md │ ├── bagua-core-c │ │ ├── Cargo.toml │ │ └── src │ │ │ └── lib.rs │ ├── bagua-core-internal │ │ ├── Cargo.toml │ │ ├── build.rs │ │ ├── cpp │ │ │ └── include │ │ │ │ ├── bagua_utils.h │ │ │ │ └── dbg.h │ │ ├── kernels │ │ │ └── bagua_kernels.cu │ │ ├── src │ │ │ ├── comm_ops │ │ │ │ ├── centralized_full_precision_synchronous.rs │ │ │ │ ├── centralized_low_precision_synchronous.rs │ │ │ │ ├── decentralized_full_precision_asynchronous.rs │ │ │ │ ├── decentralized_full_precision_synchronous.rs │ │ │ │ ├── decentralized_low_precision_synchronous.rs │ │ │ │ ├── mod.rs │ │ │ │ └── python_ffi_op.rs │ │ │ ├── communicators │ │ │ │ └── mod.rs │ │ │ ├── cuda_utils.rs │ │ │ ├── datatypes │ │ │ │ └── mod.rs │ │ │ ├── events.rs │ │ │ ├── kernels │ │ │ │ └── mod.rs │ │ │ ├── lib.rs │ │ │ ├── resource_pool │ │ │ │ └── mod.rs │ │ │ └── torch_ffi.rs │ │ └── third_party │ │ │ └── cub-1.8.0 │ │ │ ├── .cproject │ │ │ ├── .project │ │ │ ├── .settings │ │ │ ├── .gitignore │ │ │ ├── org.eclipse.cdt.codan.core.prefs │ │ │ ├── org.eclipse.cdt.core.prefs │ │ │ ├── org.eclipse.cdt.ui.prefs │ │ │ └── org.eclipse.core.runtime.prefs │ │ │ ├── CHANGE_LOG.TXT │ │ │ ├── LICENSE.TXT │ │ │ ├── README.md │ │ │ ├── common.mk │ │ │ ├── cub │ │ │ ├── agent │ │ │ │ ├── agent_histogram.cuh │ │ │ │ ├── agent_radix_sort_downsweep.cuh │ │ │ │ ├── agent_radix_sort_upsweep.cuh │ │ │ │ ├── agent_reduce.cuh │ │ │ │ ├── agent_reduce_by_key.cuh │ │ │ │ ├── agent_rle.cuh │ │ │ │ ├── agent_scan.cuh │ │ │ │ ├── agent_segment_fixup.cuh │ │ │ │ ├── agent_select_if.cuh │ │ │ │ ├── agent_spmv_orig.cuh │ │ │ │ └── single_pass_scan_operators.cuh │ │ │ ├── block │ │ │ │ ├── block_adjacent_difference.cuh │ │ │ │ ├── block_discontinuity.cuh │ │ │ │ ├── block_exchange.cuh │ │ │ │ ├── block_histogram.cuh │ │ │ │ ├── block_load.cuh │ │ │ │ ├── block_radix_rank.cuh │ │ │ │ ├── block_radix_sort.cuh │ │ │ │ ├── block_raking_layout.cuh │ │ │ │ ├── block_reduce.cuh │ │ │ │ ├── block_scan.cuh │ │ │ │ ├── block_shuffle.cuh │ │ │ │ ├── block_store.cuh │ │ │ │ └── specializations │ │ │ │ │ ├── block_histogram_atomic.cuh │ │ │ │ │ ├── block_histogram_sort.cuh │ │ │ │ │ ├── block_reduce_raking.cuh │ │ │ │ │ ├── block_reduce_raking_commutative_only.cuh │ │ │ │ │ ├── block_reduce_warp_reductions.cuh │ │ │ │ │ ├── block_scan_raking.cuh │ │ │ │ │ ├── block_scan_warp_scans.cuh │ │ │ │ │ ├── block_scan_warp_scans2.cuh │ │ │ │ │ └── block_scan_warp_scans3.cuh │ │ │ ├── cub.cuh │ │ │ ├── device │ │ │ │ ├── device_histogram.cuh │ │ │ │ ├── device_partition.cuh │ │ │ │ ├── device_radix_sort.cuh │ │ │ │ ├── device_reduce.cuh │ │ │ │ ├── device_run_length_encode.cuh │ │ │ │ ├── device_scan.cuh │ │ │ │ ├── device_segmented_radix_sort.cuh │ │ │ │ ├── device_segmented_reduce.cuh │ │ │ │ ├── device_select.cuh │ │ │ │ ├── device_spmv.cuh │ │ │ │ └── dispatch │ │ │ │ │ ├── dispatch_histogram.cuh │ │ │ │ │ ├── dispatch_radix_sort.cuh │ │ │ │ │ ├── dispatch_reduce.cuh │ │ │ │ │ ├── dispatch_reduce_by_key.cuh │ │ │ │ │ ├── dispatch_rle.cuh │ │ │ │ │ ├── dispatch_scan.cuh │ │ │ │ │ ├── dispatch_select_if.cuh │ │ │ │ │ └── dispatch_spmv_orig.cuh │ │ │ ├── grid │ │ │ │ ├── grid_barrier.cuh │ │ │ │ ├── grid_even_share.cuh │ │ │ │ ├── grid_mapping.cuh │ │ │ │ └── grid_queue.cuh │ │ │ ├── host │ │ │ │ └── mutex.cuh │ │ │ ├── iterator │ │ │ │ ├── arg_index_input_iterator.cuh │ │ │ │ ├── cache_modified_input_iterator.cuh │ │ │ │ ├── cache_modified_output_iterator.cuh │ │ │ │ ├── constant_input_iterator.cuh │ │ │ │ ├── counting_input_iterator.cuh │ │ │ │ ├── discard_output_iterator.cuh │ │ │ │ ├── tex_obj_input_iterator.cuh │ │ │ │ ├── tex_ref_input_iterator.cuh │ │ │ │ └── transform_input_iterator.cuh │ │ │ ├── thread │ │ │ │ ├── thread_load.cuh │ │ │ │ ├── thread_operators.cuh │ │ │ │ ├── thread_reduce.cuh │ │ │ │ ├── thread_scan.cuh │ │ │ │ ├── thread_search.cuh │ │ │ │ └── thread_store.cuh │ │ │ ├── util_allocator.cuh │ │ │ ├── util_arch.cuh │ │ │ ├── util_debug.cuh │ │ │ ├── util_device.cuh │ │ │ ├── util_macro.cuh │ │ │ ├── util_namespace.cuh │ │ │ ├── util_ptx.cuh │ │ │ ├── util_type.cuh │ │ │ └── warp │ │ │ │ ├── specializations │ │ │ │ ├── warp_reduce_shfl.cuh │ │ │ │ ├── warp_reduce_smem.cuh │ │ │ │ ├── warp_scan_shfl.cuh │ │ │ │ └── warp_scan_smem.cuh │ │ │ │ ├── warp_reduce.cuh │ │ │ │ └── warp_scan.cuh │ │ │ ├── eclipse code style profile.xml │ │ │ ├── examples │ │ │ ├── block │ │ │ │ ├── .gitignore │ │ │ │ ├── Makefile │ │ │ │ ├── example_block_radix_sort.cu │ │ │ │ ├── example_block_reduce.cu │ │ │ │ ├── example_block_scan.cu │ │ │ │ └── reduce_by_key.cu │ │ │ └── device │ │ │ │ ├── .gitignore │ │ │ │ ├── Makefile │ │ │ │ ├── example_device_partition_flagged.cu │ │ │ │ ├── example_device_partition_if.cu │ │ │ │ ├── example_device_radix_sort.cu │ │ │ │ ├── example_device_reduce.cu │ │ │ │ ├── example_device_scan.cu │ │ │ │ ├── example_device_select_flagged.cu │ │ │ │ ├── example_device_select_if.cu │ │ │ │ ├── example_device_select_unique.cu │ │ │ │ └── example_device_sort_find_non_trivial_runs.cu │ │ │ ├── experimental │ │ │ ├── .gitignore │ │ │ ├── Makefile │ │ │ ├── defunct │ │ │ │ ├── example_coo_spmv.cu │ │ │ │ └── test_device_seg_reduce.cu │ │ │ ├── histogram │ │ │ │ ├── histogram_cub.h │ │ │ │ ├── histogram_gmem_atomics.h │ │ │ │ └── histogram_smem_atomics.h │ │ │ ├── histogram_compare.cu │ │ │ ├── sparse_matrix.h │ │ │ ├── spmv_compare.cu │ │ │ └── spmv_script.sh │ │ │ ├── test │ │ │ ├── .gitignore │ │ │ ├── Makefile │ │ │ ├── half.h │ │ │ ├── link_a.cu │ │ │ ├── link_b.cu │ │ │ ├── link_main.cpp │ │ │ ├── mersenne.h │ │ │ ├── test_allocator.cu │ │ │ ├── test_block_histogram.cu │ │ │ ├── test_block_load_store.cu │ │ │ ├── test_block_radix_sort.cu │ │ │ ├── test_block_reduce.cu │ │ │ ├── test_block_scan.cu │ │ │ ├── test_device_histogram.cu │ │ │ ├── test_device_radix_sort.cu │ │ │ ├── test_device_reduce.cu │ │ │ ├── test_device_reduce_by_key.cu │ │ │ ├── test_device_run_length_encode.cu │ │ │ ├── test_device_scan.cu │ │ │ ├── test_device_select_if.cu │ │ │ ├── test_device_select_unique.cu │ │ │ ├── test_grid_barrier.cu │ │ │ ├── test_iterator.cu │ │ │ ├── test_util.h │ │ │ ├── test_warp_reduce.cu │ │ │ └── test_warp_scan.cu │ │ │ └── tune │ │ │ ├── .gitignore │ │ │ ├── Makefile │ │ │ └── tune_device_reduce.cu │ ├── bagua-core-py │ │ ├── Cargo.toml │ │ └── src │ │ │ └── lib.rs │ └── bagua-opentelemetry │ │ ├── Cargo.toml │ │ └── src │ │ ├── exporter │ │ ├── agent.rs │ │ └── mod.rs │ │ └── lib.rs └── bagua-net │ ├── .gitignore │ ├── Cargo.lock │ ├── Cargo.toml │ ├── README.md │ ├── cbindgen.toml │ ├── cc │ ├── Makefile │ ├── bagua_net.cc │ ├── bagua_net.h │ ├── nccl_types.h │ ├── v3 │ │ ├── nccl_net_v3.cc │ │ └── nccl_net_v3.h │ └── v4 │ │ ├── nccl_net_v4.cc │ │ └── nccl_net_v4.h │ └── src │ ├── implement │ ├── mod.rs │ ├── nthread_per_socket_backend.rs │ └── tokio_backend.rs │ ├── interface.rs │ ├── lib.rs │ └── utils.rs ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── comm └── test_communicator.py ├── contrib ├── test_cached_dataset.py ├── test_fused_optimizer.py ├── test_load_balancing_data_loader.py ├── test_store.py └── test_sync_bn.py ├── internal ├── __init__.py ├── common_utils.py ├── compressor.py ├── multi_process.py └── torch │ ├── __init__.py │ ├── _core.py │ ├── common_cuda.py │ ├── common_distributed.py │ ├── common_utils.py │ └── expecttest.py ├── pytorch_lightning ├── __init__.py ├── boring_model.py └── test_bagua_strategy.py ├── service ├── test_autotune_service.py └── test_bayesian_optimizer.py ├── test_bagua_define.py └── torch_api ├── data_parallel ├── __init__.py ├── test_async_model_average.py ├── test_bagua_ddp.py ├── test_bagua_module.py ├── test_broadcast_state.py ├── test_c10d_common.py ├── test_decentralized.py ├── test_gradient_allreduce.py ├── test_low_precision_decentralized.py └── test_multi_models.py ├── test_async_model_average.py ├── test_bagua_module.py ├── test_broadcast_state.py ├── test_decentralized.py ├── test_gradient_allreduce.py ├── test_low_precision_decentralized.py ├── test_multi_models.py ├── test_process_group.py ├── test_qadam.py └── test_utils.py /.buildkite/pipeline.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/pipeline.yml -------------------------------------------------------------------------------- /.buildkite/scripts/benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/scripts/benchmark.sh -------------------------------------------------------------------------------- /.buildkite/scripts/benchmark_master.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/scripts/benchmark_master.sh -------------------------------------------------------------------------------- /.buildkite/scripts/benchmark_worker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/scripts/benchmark_worker.sh -------------------------------------------------------------------------------- /.buildkite/scripts/example.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/scripts/example.sh -------------------------------------------------------------------------------- /.buildkite/scripts/install_bagua.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/scripts/install_bagua.sh -------------------------------------------------------------------------------- /.buildkite/scripts/run_pytest.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.buildkite/scripts/run_pytest.sh -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/issue-close-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/issue-close-app.yml -------------------------------------------------------------------------------- /.github/workflows/bagua-net-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/bagua-net-ci.yml -------------------------------------------------------------------------------- /.github/workflows/bagua-pypi-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/bagua-pypi-publish.yml -------------------------------------------------------------------------------- /.github/workflows/bagua-python-package-check.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/bagua-python-package-check.yml -------------------------------------------------------------------------------- /.github/workflows/black.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/black.yml -------------------------------------------------------------------------------- /.github/workflows/dockerhub.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/dockerhub.yml -------------------------------------------------------------------------------- /.github/workflows/flake8.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/flake8.yml -------------------------------------------------------------------------------- /.github/workflows/pytype.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/pytype.yml -------------------------------------------------------------------------------- /.github/workflows/rustfmt.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.github/workflows/rustfmt.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/.gitmodules -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/README.md -------------------------------------------------------------------------------- /bagua/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/__init__.py -------------------------------------------------------------------------------- /bagua/bagua_define.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/bagua_define.py -------------------------------------------------------------------------------- /bagua/distributed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bagua/distributed/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/distributed/launch.py -------------------------------------------------------------------------------- /bagua/distributed/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/distributed/run.py -------------------------------------------------------------------------------- /bagua/script/__init__.py: -------------------------------------------------------------------------------- 1 | from . import baguarun # noqa: F401 2 | -------------------------------------------------------------------------------- /bagua/script/bagua_sys_perf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/script/bagua_sys_perf -------------------------------------------------------------------------------- /bagua/script/baguarun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/script/baguarun.py -------------------------------------------------------------------------------- /bagua/service/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/service/__init__.py -------------------------------------------------------------------------------- /bagua/service/autotune_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/service/autotune_service.py -------------------------------------------------------------------------------- /bagua/service/autotune_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/service/autotune_system.py -------------------------------------------------------------------------------- /bagua/service/autotune_task_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/service/autotune_task_manager.py -------------------------------------------------------------------------------- /bagua/service/bayesian_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/service/bayesian_optimizer.py -------------------------------------------------------------------------------- /bagua/torch_api/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/async_model_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/async_model_average.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/base.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/bytegrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/bytegrad.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/decentralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/decentralized.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/gradient_allreduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/gradient_allreduce.py -------------------------------------------------------------------------------- /bagua/torch_api/algorithms/q_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/algorithms/q_adam.py -------------------------------------------------------------------------------- /bagua/torch_api/bucket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/bucket.py -------------------------------------------------------------------------------- /bagua/torch_api/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/checkpoint/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/checkpoint/checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/checkpoint/checkpointing.py -------------------------------------------------------------------------------- /bagua/torch_api/communication.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/communication.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/cache_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/cache_loader.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/cached_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/cached_dataset.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/fuse/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bagua/torch_api/contrib/fuse/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/fuse/optimizer.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/load_balancing_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/load_balancing_data_loader.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/sync_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/sync_batchnorm.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/utils/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/utils/redis_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/utils/redis_store.py -------------------------------------------------------------------------------- /bagua/torch_api/contrib/utils/store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/contrib/utils/store.py -------------------------------------------------------------------------------- /bagua/torch_api/data_parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/data_parallel/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/data_parallel/bagua_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/data_parallel/bagua_distributed.py -------------------------------------------------------------------------------- /bagua/torch_api/data_parallel/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/data_parallel/distributed.py -------------------------------------------------------------------------------- /bagua/torch_api/data_parallel/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/data_parallel/functional.py -------------------------------------------------------------------------------- /bagua/torch_api/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/distributed.py -------------------------------------------------------------------------------- /bagua/torch_api/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/env.py -------------------------------------------------------------------------------- /bagua/torch_api/model_parallel/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bagua/torch_api/model_parallel/moe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/model_parallel/moe/__init__.py -------------------------------------------------------------------------------- /bagua/torch_api/model_parallel/moe/experts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/model_parallel/moe/experts.py -------------------------------------------------------------------------------- /bagua/torch_api/model_parallel/moe/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/model_parallel/moe/layer.py -------------------------------------------------------------------------------- /bagua/torch_api/model_parallel/moe/sharded_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/model_parallel/moe/sharded_moe.py -------------------------------------------------------------------------------- /bagua/torch_api/model_parallel/moe/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/model_parallel/moe/utils.py -------------------------------------------------------------------------------- /bagua/torch_api/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/tensor.py -------------------------------------------------------------------------------- /bagua/torch_api/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua/torch_api/utils.py -------------------------------------------------------------------------------- /bagua/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.9.2" 2 | -------------------------------------------------------------------------------- /bagua_core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua_core/__init__.py -------------------------------------------------------------------------------- /bagua_core/_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua_core/_environment.py -------------------------------------------------------------------------------- /bagua_core/bagua_install_deps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/bagua_core/bagua_install_deps.py -------------------------------------------------------------------------------- /cliff.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/cliff.toml -------------------------------------------------------------------------------- /docker/Dockerfile.manylinux-cuda102: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.manylinux-cuda102 -------------------------------------------------------------------------------- /docker/Dockerfile.manylinux-cuda111: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.manylinux-cuda111 -------------------------------------------------------------------------------- /docker/Dockerfile.manylinux-cuda113: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.manylinux-cuda113 -------------------------------------------------------------------------------- /docker/Dockerfile.manylinux-cuda115: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.manylinux-cuda115 -------------------------------------------------------------------------------- /docker/Dockerfile.manylinux-cuda116: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.manylinux-cuda116 -------------------------------------------------------------------------------- /docker/Dockerfile.manylinux-cuda117: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.manylinux-cuda117 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-1.10.0-cuda11.3-cudnn8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-1.10.0-cuda11.3-cudnn8 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-1.11.0-cuda11.3-cudnn8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-1.11.0-cuda11.3-cudnn8 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-1.12.0-cuda11.3-cudnn8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-1.12.0-cuda11.3-cudnn8 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-1.13.0-cuda11.6-cudnn8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-1.13.0-cuda11.6-cudnn8 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-1.9.0-cuda10.2-cudnn7: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-1.9.0-cuda10.2-cudnn7 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-1.9.0-cuda11.1-cudnn8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-1.9.0-cuda11.1-cudnn8 -------------------------------------------------------------------------------- /docker/Dockerfile.pytorch-2.0.0-cuda11.7-cudnn8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docker/Dockerfile.pytorch-2.0.0-cuda11.7-cudnn8 -------------------------------------------------------------------------------- /docs/_autoapi_templates/index.rst: -------------------------------------------------------------------------------- 1 | ../index.rst -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/attribute.rst: -------------------------------------------------------------------------------- 1 | {% extends "python/data.rst" %} 2 | -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/_autoapi_templates/python/class.rst -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/_autoapi_templates/python/data.rst -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/exception.rst: -------------------------------------------------------------------------------- 1 | {% extends "python/class.rst" %} 2 | -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/_autoapi_templates/python/function.rst -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/method.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/_autoapi_templates/python/method.rst -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/_autoapi_templates/python/module.rst -------------------------------------------------------------------------------- /docs/_autoapi_templates/python/package.rst: -------------------------------------------------------------------------------- 1 | {% extends "python/module.rst" %} 2 | -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/doc-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/doc-requirements.txt -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/docs/index.rst -------------------------------------------------------------------------------- /examples/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/LICENSE -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/benchmark/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/benchmark/README.md -------------------------------------------------------------------------------- /examples/benchmark/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/benchmark/requirements.txt -------------------------------------------------------------------------------- /examples/benchmark/synthetic_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/benchmark/synthetic_benchmark.py -------------------------------------------------------------------------------- /examples/communication_primitives/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/communication_primitives/README.md -------------------------------------------------------------------------------- /examples/communication_primitives/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/communication_primitives/main.py -------------------------------------------------------------------------------- /examples/communication_primitives/requirements.txt: -------------------------------------------------------------------------------- 1 | bagua>=0.7 2 | torch 3 | -------------------------------------------------------------------------------- /examples/elastic_training/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/elastic_training/README.md -------------------------------------------------------------------------------- /examples/elastic_training/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/elastic_training/main.py -------------------------------------------------------------------------------- /examples/elastic_training/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/elastic_training/requirements.txt -------------------------------------------------------------------------------- /examples/imagenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/imagenet/README.md -------------------------------------------------------------------------------- /examples/imagenet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/imagenet/main.py -------------------------------------------------------------------------------- /examples/imagenet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/imagenet/requirements.txt -------------------------------------------------------------------------------- /examples/mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/mnist/README.md -------------------------------------------------------------------------------- /examples/mnist/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/mnist/main.py -------------------------------------------------------------------------------- /examples/mnist/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/mnist/requirements.txt -------------------------------------------------------------------------------- /examples/moe/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/moe/README.md -------------------------------------------------------------------------------- /examples/moe/mnist_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/moe/mnist_main.py -------------------------------------------------------------------------------- /examples/moe/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/moe/requirements.txt -------------------------------------------------------------------------------- /examples/squad/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/squad/README.md -------------------------------------------------------------------------------- /examples/squad/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/examples/squad/main.py -------------------------------------------------------------------------------- /examples/squad/requirements.txt: -------------------------------------------------------------------------------- 1 | bagua>=0.6 2 | torch 3 | tqdm 4 | transformers<=4.11.3 5 | tensorboardX 6 | -------------------------------------------------------------------------------- /figures/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/figures/logo.png -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/install.sh -------------------------------------------------------------------------------- /install_master.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/install_master.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/pyproject.toml -------------------------------------------------------------------------------- /rust/bagua-core/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/.gitignore -------------------------------------------------------------------------------- /rust/bagua-core/Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/Cargo.lock -------------------------------------------------------------------------------- /rust/bagua-core/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/Cargo.toml -------------------------------------------------------------------------------- /rust/bagua-core/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/README.md -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-c/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-c/Cargo.toml -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-c/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-c/src/lib.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/Cargo.toml -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/build.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/build.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/cpp/include/bagua_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/cpp/include/bagua_utils.h -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/cpp/include/dbg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/cpp/include/dbg.h -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/kernels/bagua_kernels.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/kernels/bagua_kernels.cu -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/comm_ops/centralized_full_precision_synchronous.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/comm_ops/mod.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/comm_ops/python_ffi_op.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/comm_ops/python_ffi_op.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/communicators/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/communicators/mod.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/cuda_utils.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/cuda_utils.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/datatypes/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/datatypes/mod.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/events.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/events.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/kernels/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/kernels/mod.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/lib.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/resource_pool/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/resource_pool/mod.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/src/torch_ffi.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-internal/src/torch_ffi.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.cproject: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/.settings/.gitignore: -------------------------------------------------------------------------------- 1 | /language.settings.xml 2 | -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/.settings/org.eclipse.cdt.codan.core.prefs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/CHANGE_LOG.TXT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/cub/agent/agent_histogram.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/cub.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/device/device_histogram.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/util_allocator.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/warp/specializations/warp_reduce_shfl.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/cub/warp/warp_reduce.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/eclipse code style profile.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/examples/block/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- 1 | /bin 2 | -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/experimental/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/defunct/example_coo_spmv.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/experimental/histogram_compare.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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/test/.gitignore: -------------------------------------------------------------------------------- 1 | /bin 2 | /link_main.obj 3 | /dummy/ 4 | -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/test/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- 1 | /bin 2 | -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-internal/third_party/cub-1.8.0/tune/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/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: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-py/Cargo.toml -------------------------------------------------------------------------------- /rust/bagua-core/bagua-core-py/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-core-py/src/lib.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-opentelemetry/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-opentelemetry/Cargo.toml -------------------------------------------------------------------------------- /rust/bagua-core/bagua-opentelemetry/src/exporter/agent.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-opentelemetry/src/exporter/agent.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-opentelemetry/src/exporter/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-opentelemetry/src/exporter/mod.rs -------------------------------------------------------------------------------- /rust/bagua-core/bagua-opentelemetry/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-core/bagua-opentelemetry/src/lib.rs -------------------------------------------------------------------------------- /rust/bagua-net/.gitignore: -------------------------------------------------------------------------------- 1 | /target 2 | -------------------------------------------------------------------------------- /rust/bagua-net/Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/Cargo.lock -------------------------------------------------------------------------------- /rust/bagua-net/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/Cargo.toml -------------------------------------------------------------------------------- /rust/bagua-net/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/README.md -------------------------------------------------------------------------------- /rust/bagua-net/cbindgen.toml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rust/bagua-net/cc/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/Makefile -------------------------------------------------------------------------------- /rust/bagua-net/cc/bagua_net.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/bagua_net.cc -------------------------------------------------------------------------------- /rust/bagua-net/cc/bagua_net.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/bagua_net.h -------------------------------------------------------------------------------- /rust/bagua-net/cc/nccl_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/nccl_types.h -------------------------------------------------------------------------------- /rust/bagua-net/cc/v3/nccl_net_v3.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/v3/nccl_net_v3.cc -------------------------------------------------------------------------------- /rust/bagua-net/cc/v3/nccl_net_v3.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/v3/nccl_net_v3.h -------------------------------------------------------------------------------- /rust/bagua-net/cc/v4/nccl_net_v4.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/v4/nccl_net_v4.cc -------------------------------------------------------------------------------- /rust/bagua-net/cc/v4/nccl_net_v4.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/cc/v4/nccl_net_v4.h -------------------------------------------------------------------------------- /rust/bagua-net/src/implement/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/src/implement/mod.rs -------------------------------------------------------------------------------- /rust/bagua-net/src/implement/nthread_per_socket_backend.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/src/implement/nthread_per_socket_backend.rs -------------------------------------------------------------------------------- /rust/bagua-net/src/implement/tokio_backend.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/src/implement/tokio_backend.rs -------------------------------------------------------------------------------- /rust/bagua-net/src/interface.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/src/interface.rs -------------------------------------------------------------------------------- /rust/bagua-net/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/src/lib.rs -------------------------------------------------------------------------------- /rust/bagua-net/src/utils.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/rust/bagua-net/src/utils.rs -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [pytype] 2 | inputs = bagua 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/comm/test_communicator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/comm/test_communicator.py -------------------------------------------------------------------------------- /tests/contrib/test_cached_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/contrib/test_cached_dataset.py -------------------------------------------------------------------------------- /tests/contrib/test_fused_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/contrib/test_fused_optimizer.py -------------------------------------------------------------------------------- /tests/contrib/test_load_balancing_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/contrib/test_load_balancing_data_loader.py -------------------------------------------------------------------------------- /tests/contrib/test_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/contrib/test_store.py -------------------------------------------------------------------------------- /tests/contrib/test_sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/contrib/test_sync_bn.py -------------------------------------------------------------------------------- /tests/internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/internal/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/common_utils.py -------------------------------------------------------------------------------- /tests/internal/compressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/compressor.py -------------------------------------------------------------------------------- /tests/internal/multi_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/multi_process.py -------------------------------------------------------------------------------- /tests/internal/torch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/internal/torch/_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/torch/_core.py -------------------------------------------------------------------------------- /tests/internal/torch/common_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/torch/common_cuda.py -------------------------------------------------------------------------------- /tests/internal/torch/common_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/torch/common_distributed.py -------------------------------------------------------------------------------- /tests/internal/torch/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/torch/common_utils.py -------------------------------------------------------------------------------- /tests/internal/torch/expecttest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/internal/torch/expecttest.py -------------------------------------------------------------------------------- /tests/pytorch_lightning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/pytorch_lightning/boring_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/pytorch_lightning/boring_model.py -------------------------------------------------------------------------------- /tests/pytorch_lightning/test_bagua_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/pytorch_lightning/test_bagua_strategy.py -------------------------------------------------------------------------------- /tests/service/test_autotune_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/service/test_autotune_service.py -------------------------------------------------------------------------------- /tests/service/test_bayesian_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/service/test_bayesian_optimizer.py -------------------------------------------------------------------------------- /tests/test_bagua_define.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/test_bagua_define.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_async_model_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_async_model_average.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_bagua_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_bagua_ddp.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_bagua_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_bagua_module.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_broadcast_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_broadcast_state.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_c10d_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_c10d_common.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_decentralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_decentralized.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_gradient_allreduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_gradient_allreduce.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_low_precision_decentralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_low_precision_decentralized.py -------------------------------------------------------------------------------- /tests/torch_api/data_parallel/test_multi_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/data_parallel/test_multi_models.py -------------------------------------------------------------------------------- /tests/torch_api/test_async_model_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_async_model_average.py -------------------------------------------------------------------------------- /tests/torch_api/test_bagua_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_bagua_module.py -------------------------------------------------------------------------------- /tests/torch_api/test_broadcast_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_broadcast_state.py -------------------------------------------------------------------------------- /tests/torch_api/test_decentralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_decentralized.py -------------------------------------------------------------------------------- /tests/torch_api/test_gradient_allreduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_gradient_allreduce.py -------------------------------------------------------------------------------- /tests/torch_api/test_low_precision_decentralized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_low_precision_decentralized.py -------------------------------------------------------------------------------- /tests/torch_api/test_multi_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_multi_models.py -------------------------------------------------------------------------------- /tests/torch_api/test_process_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_process_group.py -------------------------------------------------------------------------------- /tests/torch_api/test_qadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_qadam.py -------------------------------------------------------------------------------- /tests/torch_api/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaguaSys/bagua/HEAD/tests/torch_api/test_utils.py --------------------------------------------------------------------------------