傅剑寒 7ebdf48ac5 add cast and op_functor for cuda build-in types (#5546)

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