diff --git a/include/cutlass/arch/memory.h b/include/cutlass/arch/memory.h index 50f3d46a..db9ad739 100644 --- a/include/cutlass/arch/memory.h +++ b/include/cutlass/arch/memory.h @@ -36,6 +36,7 @@ #include "cutlass/cutlass.h" #include "cutlass/arch/cache_operation.h" +#include "cutlass/platform/platform.h" namespace cutlass { namespace arch { diff --git a/include/cutlass/numeric_types.h b/include/cutlass/numeric_types.h index 78313350..cb7c2087 100644 --- a/include/cutlass/numeric_types.h +++ b/include/cutlass/numeric_types.h @@ -44,6 +44,7 @@ #pragma once #include "cutlass/cutlass.h" +#include "cutlass/platform/platform.h" #include "cutlass/numeric_size.h" /////////////////////////////////////////////////////////////////////////////////////////////////