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 dde03ce944 add async ops excute for pynative 5 years ago
..
distribution support host reduce scatter and allgather 5 years ago
mpi support host reduce scatter and allgather 5 years ago
blocking_queue.cc gpu queue support multi-inputs 5 years ago
blocking_queue.h gpu queue support multi-inputs 5 years ago
cuda_common.h gpu change compute capacity strategy 5 years ago
cuda_driver.cc change the default memory copy way to async 5 years ago
cuda_driver.h add memory copy module 5 years ago
gpu_buffer_mgr.cc gpu iterator weak ref opt 5 years ago
gpu_buffer_mgr.h gpu iterator weak ref opt 5 years ago
gpu_common.h initial version 6 years ago
gpu_device_address.cc add async ops excute for pynative 5 years ago
gpu_device_address.h support mix target 5 years ago
gpu_device_manager.cc change the default memory copy way to async 5 years ago
gpu_device_manager.h change the default memory copy way to async 5 years ago
gpu_kernel_build.cc change directory of akg cuda kernel 5 years ago
gpu_kernel_build.h initial version 6 years ago
gpu_kernel_runtime.cc fix resource release bug of memory swap 5 years ago
gpu_kernel_runtime.h change gpu kernel runtime to support memory swap 5 years ago
gpu_memory_allocator.cc update clang format rule 5 years ago
gpu_memory_allocator.h update clang format rule 5 years ago
gpu_memory_copy_manager.cc add memory copy module 5 years ago
gpu_memory_copy_manager.h add memory copy module 5 years ago
gpu_memory_manager.cc gpu optimize some return values of dynamic memory pool 5 years ago
gpu_memory_manager.h gpu dynamic memory pool supports multi-allReduce 5 years ago
gpu_stream_assign.cc Fix result error when calling AllReduce serially. 5 years ago
gpu_stream_assign.h Fix result error when calling AllReduce serially. 5 years ago
kernel_info_setter.cc gpu update shape infer 5 years ago
kernel_info_setter.h update clang format rule 5 years ago
readme.md initial version 6 years ago