| .. |
|
distribution
|
add dump pb and nccl protected
|
2021-07-02 11:25:43 +08:00 |
|
mpi
|
code warning clean
|
2020-09-21 09:44:39 +08:00 |
|
blocking_queue.cc
|
fix codecheck and pclint
|
2021-05-26 14:07:29 +08:00 |
|
blocking_queue.h
|
add input data type check for ps cache mode
|
2021-02-07 12:35:04 +08:00 |
|
cuda_common.h
|
add more dtypes support for gatherdgrad and other bugfix
|
2020-11-30 14:00:54 +08:00 |
|
cuda_driver.cc
|
fix an allreduce bug with two streams sync problem
|
2021-05-10 10:00:58 +08:00 |
|
cuda_driver.h
|
fix an allreduce bug with two streams sync problem
|
2021-05-10 10:00:58 +08:00 |
|
cuda_env_checker.cc
|
search nvcc in entire PATH
|
2020-09-04 15:55:48 +08:00 |
|
cuda_env_checker.h
|
search nvcc in entire PATH
|
2020-09-04 15:55:48 +08:00 |
|
gpu_bucket.cc
|
unify runtime for PyNative distributed mode
|
2021-05-20 11:39:14 +08:00 |
|
gpu_bucket.h
|
unify runtime for PyNative distributed mode
|
2021-05-20 11:39:14 +08:00 |
|
gpu_buffer_mgr.cc
|
add error log when set device id failed
|
2021-03-25 14:58:03 +08:00 |
|
gpu_buffer_mgr.h
|
add push opt logic
|
2020-12-15 15:35:42 +08:00 |
|
gpu_common.h
|
Unify runtime support profiling
|
2021-05-24 20:04:11 +08:00 |
|
gpu_device_address.cc
|
fix device bug of the input empty
|
2021-07-06 14:25:45 +08:00 |
|
gpu_device_address.h
|
support group conv2d in pynative
|
2021-06-26 11:48:06 +08:00 |
|
gpu_device_manager.cc
|
fix the test case of CPU dump
|
2021-07-12 22:58:34 +08:00 |
|
gpu_device_manager.h
|
fix an allreduce bug with two streams sync problem
|
2021-05-10 10:00:58 +08:00 |
|
gpu_event.cc
|
PyNative AllReduce Bucket
|
2021-03-03 15:47:18 +08:00 |
|
gpu_event.h
|
PyNative AllReduce Bucket
|
2021-03-03 15:47:18 +08:00 |
|
gpu_kernel_build.cc
|
using kernel pool to share the compiling results when running on multi
|
2021-08-02 15:25:17 +08:00 |
|
gpu_kernel_build.h
|
add hardware abstract layer
|
2021-03-19 11:28:53 +08:00 |
|
gpu_kernel_runtime.cc
|
using kernel pool to share the compiling results when running on multi
|
2021-08-02 15:25:17 +08:00 |
|
gpu_kernel_runtime.h
|
support group conv2d in pynative
|
2021-06-26 11:48:06 +08:00 |
|
gpu_launch_kernel.cc
|
add hardware abstract layer
|
2021-03-19 11:28:53 +08:00 |
|
gpu_launch_kernel.h
|
add op atomic clean to clear input addr in launch allreduce
|
2021-03-25 11:19:23 +08:00 |
|
gpu_launch_mul.cc
|
add op_mul fusion based on allreduce fusion in pynative mode
|
2021-03-15 21:12:14 +08:00 |
|
gpu_launch_mul.h
|
add op atomic clean to clear input addr in launch allreduce
|
2021-03-25 11:19:23 +08:00 |
|
gpu_memory_allocator.cc
|
fix codecheck and pclint
|
2021-05-26 14:07:29 +08:00 |
|
gpu_memory_allocator.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
gpu_memory_copy_manager.cc
|
fix an allreduce bug with two streams sync problem
|
2021-05-10 10:00:58 +08:00 |
|
gpu_memory_copy_manager.h
|
fix an allreduce bug with two streams sync problem
|
2021-05-10 10:00:58 +08:00 |
|
gpu_memory_manager.cc
|
add the continue memory alloc of communication kernel for actor runtime
|
2021-04-26 10:30:25 +08:00 |
|
gpu_memory_manager.h
|
profiler memory
|
2021-01-19 10:21:43 +08:00 |
|
gpu_stream_assign.cc
|
actor runtimie supports allreduce multi-stream
|
2021-05-31 18:20:14 +08:00 |
|
gpu_stream_assign.h
|
actor runtimie supports allreduce multi-stream
|
2021-05-31 18:20:14 +08:00 |
|
kernel_info_setter.cc
|
unify runtime support pynative hook
|
2021-06-15 22:10:17 +08:00 |
|
kernel_info_setter.h
|
1.Optimize bias add grad kernel
|
2021-06-10 00:28:26 +08:00 |
|
queue_common.h
|
add trace for gpu error/excpt log
|
2020-12-04 14:03:36 +08:00 |
|
readme.md
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
trt_loader.cc
|
trt operator
|
2021-05-29 14:49:44 +08:00 |
|
trt_loader.h
|
gpu inference mixed precision
|
2021-04-25 09:53:47 +08:00 |
d293c5eb26