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.
 
 
 
 
 
 
John Tzanakakis b291861e85 LoadTensors was using a 4-byte size field instead of an 8-byte size field 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 improve perf, keep consistent tensor state, fix recheck, check weights at step end 5 years ago
gpu_device_address.h load inputs before suspending execution in dbg 5 years ago
gpu_device_manager.cc gpu all-reduce memory alloc fixed 5 years ago
gpu_device_manager.h gpu all-reduce memory alloc fixed 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 LoadTensors was using a 4-byte size field instead of an 8-byte size field 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