ColossalAI/extensions/csrc/cuda/utils
2024-04-08 11:38:05 +08:00
..
gpu_launch_config.h add implementatino for GetGPULaunchConfig1D 2024-03-14 11:13:40 +08:00
micros.h add cast and op_functor for cuda build-in types (#5546) 2024-04-08 11:38:05 +08:00
nvgpu_dev_info.h add implementatino for GetGPULaunchConfig1D 2024-03-14 11:13:40 +08:00
vec_type_traits.h add vec_type_trait implementation (#5473) 2024-03-19 18:36:40 +08:00
vector_copy_utils.h [Inference]Support FP16/BF16 Flash Attention 2 And Add high_precision Flag To Rotary Embedding (#5461) 2024-03-25 13:40:34 +08:00