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/kernel
Jun Gao dce05da535
fix thrust-transform-reduce error (#5078)
1 year ago
..
cuda_native fix thrust-transform-reduce error (#5078) 1 year ago
jit [misc] update pre-commit and run all files (#4752) 1 year ago
triton [Kernels]added flash-decoidng of triton (#5063) 1 year ago
__init__.py [hotfix] Fix import error: colossal.kernel without triton installed (#4722) 1 year ago
op_builder