cub
|
optimize cumsum with unroll
|
2021-09-02 17:22:46 +08:00 |
cublas
|
polish matmul
|
2022-03-18 16:21:34 +08:00 |
cudnn
|
add fp16 support
|
2022-03-15 17:45:39 +08:00 |
cufft
|
support float64 fft
|
2022-03-26 23:20:16 -04:00 |
curand
|
rename wrapper
|
2021-11-01 12:01:01 +08:00 |
cutt/ops
|
add acl backend
|
2022-03-21 17:48:50 +08:00 |
inc
|
add acl backend
|
2022-03-21 17:48:50 +08:00 |
nccl
|
rename wrapper
|
2021-11-01 12:01:01 +08:00 |
src
|
data gz and register_hook and requires_grads_
|
2021-07-26 20:58:15 +08:00 |