- 30 12月, 2021 1 次提交
-
-
由 zhangkaihuo 提交于
将cuSparse的handle与DeviceContext进行绑定，避免op中进行创建和销毁 添加对cuSparse中dense和sparse转换的API进行封装 添加对封装的API的单测
-
- 27 12月, 2021 1 次提交
-
-
由 sneaxiy 提交于
-
- 17 12月, 2021 1 次提交
-
-
由 From00 提交于
* Get GPU BasePtr from CUDA allocation * Fix compile error for ROCm * Add BasePtr function for IPUPlace in naive_best_fit_allocator.cc * Add alignment for BuddyAllocator * Set address alignment of BuddyAllocator to 32 bytes * Fix CI error * Remove code for naive_best_fit strategy
-
- 10 12月, 2021 1 次提交
-
-
由 sneaxiy 提交于
-
- 09 12月, 2021 1 次提交
-
-
由 sneaxiy 提交于
* fix cuda atomicAdd for FP16 * try to fix ci
-
- 03 12月, 2021 1 次提交
-
-
由 ronnywang 提交于
* refine structure for cuda and rocm * update * update * update * update
-