From 29801e348adc609973cc1e4fba659b3d58384f8e Mon Sep 17 00:00:00 2001 From: Stepan Tezyunichev Date: Fri, 10 Mar 2023 07:24:47 +0300 Subject: [PATCH] Hide streams and typinfo from nvrtc (#853) * Hide streams and typinfo from nvrtc * Use __CUDACC_RTC__ instead CUDA_ARCH for guard --- include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h b/include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h index 239ced7a..3f8d7a8d 100644 --- a/include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h +++ b/include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h @@ -43,10 +43,11 @@ #include "cutlass/conv/conv3d_problem_size.h" #include "cutlass/gemm/threadblock/index_remat.h" +#if !defined(__CUDACC_RTC__) #include #include "cutlass/core_io.h" #include "cutlass/trace.h" - +#endif