diff --git a/cute_kernels/include/dtypes/all.h b/cute_kernels/include/dtypes/all.h index 149576cb..71c64548 100644 --- a/cute_kernels/include/dtypes/all.h +++ b/cute_kernels/include/dtypes/all.h @@ -1,7 +1,10 @@ #pragma once -#include "bf16.h" -#include "fp16.h" -#include "fp32.h" -#include "fp64.h" -#include "uint32.h" +#include "alias.h" +#include "bf16.cuh" +#include "common.h" +#include "fp16.cuh" +#include "fp32.cuh" +#include "fp64.cuh" +#include "uint32.cuh" +#include "uint64.cuh" diff --git a/cute_kernels/include/dtypes/uint32.h b/cute_kernels/include/dtypes/uint32.cuh similarity index 100% rename from cute_kernels/include/dtypes/uint32.h rename to cute_kernels/include/dtypes/uint32.cuh