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.
 
 
 
 
 
 
i-robot 438169e0b9 !22365 RDR adapts for CPU dynamic memory allocation 4 years ago
..
distribution [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
mpi code warning clean 5 years ago
blocking_queue.cc fix codecheck and pclint 4 years ago
blocking_queue.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
cuda_common.h add more dtypes support for gatherdgrad and other bugfix 5 years ago
cuda_driver.cc fix an allreduce bug with two streams sync problem 4 years ago
cuda_driver.h fix an allreduce bug with two streams sync problem 4 years ago
cuda_env_checker.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
cuda_env_checker.h search nvcc in entire PATH 5 years ago
gpu_bucket.cc unify runtime for PyNative distributed mode 4 years ago
gpu_bucket.h unify runtime for PyNative distributed mode 4 years ago
gpu_buffer_mgr.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
gpu_buffer_mgr.h add push opt logic 5 years ago
gpu_common.h add error string for curandStatus 4 years ago
gpu_device_address.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
gpu_device_address.h support group conv2d in pynative 4 years ago
gpu_device_manager.cc fix the test case of CPU dump 4 years ago
gpu_device_manager.h fix an allreduce bug with two streams sync problem 4 years ago
gpu_event.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
gpu_event.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
gpu_kernel_build.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
gpu_kernel_build.h add hardware abstract layer 5 years ago
gpu_kernel_runtime.cc RDR adapts for CPU dynamic memory allocation 4 years ago
gpu_kernel_runtime.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
gpu_launch_kernel.cc add hardware abstract layer 5 years ago
gpu_launch_kernel.h add op atomic clean to clear input addr in launch allreduce 5 years ago
gpu_launch_mul.cc add op_mul fusion based on allreduce fusion in pynative mode 5 years ago
gpu_launch_mul.h add op atomic clean to clear input addr in launch allreduce 5 years ago
gpu_memory_allocator.cc fix codecheck and pclint 4 years ago
gpu_memory_allocator.h Unified code style 5 years ago
gpu_memory_copy_manager.cc fix an allreduce bug with two streams sync problem 4 years ago
gpu_memory_copy_manager.h fix an allreduce bug with two streams sync problem 4 years ago
gpu_memory_manager.cc add the continue memory alloc of communication kernel for actor runtime 4 years ago
gpu_memory_manager.h profiler memory 5 years ago
gpu_stream_assign.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
gpu_stream_assign.h actor runtimie supports allreduce multi-stream 4 years ago
kernel_info_setter.cc add graph kernel userdefine op support 4 years ago
kernel_info_setter.h 1.Optimize bias add grad kernel 4 years ago
queue_common.h add trace for gpu error/excpt log 5 years ago
readme.md mindspore path adjust 5 years ago
trt_loader.cc trt operator 4 years ago
trt_loader.h gpu inference mixed precision 4 years ago