kernels/attention/attention_dtypes.h (5 lines of code) (raw):

#pragma once #include "attention_generic.cuh" #include "dtype_float16.cuh" #include "dtype_float32.cuh" #include "dtype_bfloat16.cuh"