From 8e7d9f483d188fd6d7fa5c5946ae6af94975a7d8 Mon Sep 17 00:00:00 2001 From: lzw <46190841+Ghost-LZW@users.noreply.github.com> Date: Wed, 10 Apr 2024 02:15:48 +0800 Subject: [PATCH] add missing header for size_t in `numeric_types.h` (#1420) * add missing header for size_t in `numeric_types.h` * make nvrtc happy * add missing header for int types in `cutlass/arch/memory.h` --------- Co-authored-by: Haicheng Wu --- include/cutlass/arch/memory.h | 1 + include/cutlass/numeric_types.h | 1 + 2 files changed, 2 insertions(+) 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" /////////////////////////////////////////////////////////////////////////////////////////////////