mpi4jax:JAX阵列的零拷贝MPI通信，用于Python中的涡轮增压HPC应用程序
需积分: 9 144 浏览量
2021-04-05
17:07:25
上传
评论
收藏 4.56MB ZIP 举报
mpi4jax-master.zip （80个子文件） 
mpi4jax-master mpi4jax 
_deprecations.py 1KB _version.py 18KB __init__.py 797B _src xla_bridge mpi_xla_bridge_cpu.pyx 8KB mpi_xla_bridge.pyx 7KB mpi_xla_bridge.pxd 2KB __init__.py 512B mpi_xla_bridge_gpu.pyx 27KB cuda_runtime_api.pxd 626B comm.py 207B decorators.py 3KB utils.py 2KB __init__.py 2KB collective_ops allreduce.py 6KB alltoall.py 5KB sendrecv.py 7KB scatter.py 5KB reduce.py 5KB scan.py 4KB recv.py 5KB gather.py 5KB __init__.py 0B allgather.py 5KB bcast.py 5KB send.py 4KB validation.py 3KB flush.py 264B conf travis-install-mpi.sh 4KB ompi_rootenv.patch 1KB install-cuda-ubuntu.sh 5KB MANIFEST.in 220B README.rst 3KB .gitattributes 47B .flake8 207B .github workflows python-publish.yml 1KB covecov-coverage.yml 1KB build-gpu-ext.yml 2KB pre-commit.yml 241B mpi-tests.yml 2KB build-docs.yml 794B versioneer.py 69KB examples shallow_water.py 16KB setup.cfg 162B LICENSE.md 1KB setup.py 6KB .pre-commit-config.yaml 255B docs make.bat 795B shallow-water.rst 4KB conf.py 2KB usage.rst 3KB installation.rst 3KB shallow-water-source.rst 241B environment.yml 174B sharp-bits.rst 4KB index.rst 235B Makefile 634B api.rst 601B _static 
shallow-water.mp4 4.47MB tests test_invalid_jaxlib.py 913B test_examples.py 527B test_flush.py 142B test_deprecations.py 2KB test_decorators.py 1KB conftest.py 376B collective_ops test_sendrecv.py 3KB test_allreduce_matvec.py 5KB test_scatter.py 1KB test_bcast.py 1KB test_scan.py 884B test_reduce.py 1KB test_gather.py 1KB test_allgather.py 916B test_allreduce.py 5KB test_send_and_recv.py 4KB test_alltoall.py 834B test_common.py 3KB test_validation.py 2KB .readthedocs.yml 193B .gitignore 2KB pyproject.toml 646B资源评论
