ColossalAI/colossalai/utils/multi_tensor_apply
Frank Lee 40d376c566
[setup] support pre-build and jit-build of cuda kernels (#2374)
* [setup] support pre-build and jit-build of cuda kernels

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
2023-01-06 20:50:26 +08:00
..
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
multi_tensor_apply.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00