You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/utils/multi_tensor_apply
Frank Lee 40d376c566
[setup] support pre-build and jit-build of cuda kernels (#2374)
2 years ago
..
__init__.py
multi_tensor_apply.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago