You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
chujinjin ade9a82c2b fix device memory leak 5 years ago
..
distribution GPU supports p2p nccl interfaces 5 years ago
mpi code warning clean 5 years ago
blocking_queue.cc add push opt logic 5 years ago
blocking_queue.h add push opt logic 5 years ago
cuda_common.h add more dtypes support for gatherdgrad and other bugfix 5 years ago
cuda_driver.cc gpu all-reduce memory alloc fixed 5 years ago
cuda_driver.h gpu all-reduce memory alloc fixed 5 years ago
cuda_env_checker.cc search nvcc in entire PATH 5 years ago
cuda_env_checker.h search nvcc in entire PATH 5 years ago
gpu_buffer_mgr.cc add push opt logic 5 years ago
gpu_buffer_mgr.h add push opt logic 5 years ago
gpu_common.h ps cache data process thread support exit when exceptions occur 5 years ago
gpu_device_address.cc fix device memory leak 5 years ago
gpu_device_address.h fix device memory leak 5 years ago
gpu_device_manager.cc Fix GPU sync stream Segmentation fault 5 years ago
gpu_device_manager.h Fix GPU sync stream Segmentation fault 5 years ago
gpu_kernel_build.cc gpu suppor dynamic shape 5 years ago
gpu_kernel_build.h Unified code style 5 years ago
gpu_kernel_runtime.cc !10021 Fixed double stutter issue at the end of step for debugger 5 years ago
gpu_kernel_runtime.h pynative-support-dynamic-op-run-in-gpu 5 years ago
gpu_memory_allocator.cc Refactor ms_context implementation 5 years ago
gpu_memory_allocator.h Unified code style 5 years ago
gpu_memory_copy_manager.cc refine GPU memory swap performance 5 years ago
gpu_memory_copy_manager.h refine GPU memory swap performance 5 years ago
gpu_memory_manager.cc gpu all-reduce memory alloc fixed 5 years ago
gpu_memory_manager.h gpu all-reduce memory alloc fixed 5 years ago
gpu_stream_assign.cc Return false if Recv node is not found in gpu_stream_assign.cc 5 years ago
gpu_stream_assign.h fix_consecutive_allreduce_bug 5 years ago
kernel_info_setter.cc add reduce precision in pynative mode 5 years ago
kernel_info_setter.h Add relu6 gpu 5 years ago
queue_common.h add trace for gpu error/excpt log 5 years ago
readme.md mindspore path adjust 5 years ago