Hide streams and typinfo from nvrtc (#853)

* Hide streams and typinfo from nvrtc * Use __CUDACC_RTC__ instead CUDA_ARCH for guard
2023-03-10 07:24:47 +03:00 · 2023-03-10 07:24:47 +03:00 · 29801e348a
commit 29801e348a
parent 7e370c9637
1 changed files with 2 additions and 1 deletions
--- 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 <iostream>
 #include "cutlass/core_io.h"
 #include "cutlass/trace.h"
-
+#endif