ColossalAI/colossalai/kernel
digger yu 8abc87798f
fix Tensor is not defined (#4129)
2023-07-03 17:10:18 +08:00
..
cuda_native [bf16] add bf16 support (#3882) 2023-06-05 15:58:31 +08:00
jit fix Tensor is not defined (#4129) 2023-07-03 17:10:18 +08:00
__init__.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
op_builder [builder] reconfig op_builder for pypi install (#2314) 2023-01-04 16:32:32 +08:00