From f02913c34e9c0ad991b30abf756f95a2329c5bb0 Mon Sep 17 00:00:00 2001 From: MaxAkaAltmer Date: Wed, 23 Oct 2024 19:55:59 +0300 Subject: [PATCH] Include of regular_tile_iterator.h fixed for NVRTC (#1765) * Include of regular_tile_iterator.h fixed for NVRTC * More include fixed for NVRTC --- include/cutlass/epilogue/threadblock/default_thread_map_simt.h | 2 +- .../cutlass/epilogue/threadblock/default_thread_map_tensor_op.h | 2 +- .../epilogue/threadblock/default_thread_map_volta_tensor_op.h | 2 +- .../epilogue/threadblock/default_thread_map_wmma_tensor_op.h | 2 +- .../transform/threadblock/regular_tile_iterator_pitch_linear.h | 2 +- .../regular_tile_iterator_pitch_linear_2dthreadtile.h | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/include/cutlass/epilogue/threadblock/default_thread_map_simt.h b/include/cutlass/epilogue/threadblock/default_thread_map_simt.h index a5559f67..2092caf4 100644 --- a/include/cutlass/epilogue/threadblock/default_thread_map_simt.h +++ b/include/cutlass/epilogue/threadblock/default_thread_map_simt.h @@ -35,7 +35,7 @@ #pragma once -#include "predicated_tile_iterator.h" +#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h" #include "cutlass/gemm/gemm.h" ///////////////////////////////////////////////////////////////////////////////////////////////// diff --git a/include/cutlass/epilogue/threadblock/default_thread_map_tensor_op.h b/include/cutlass/epilogue/threadblock/default_thread_map_tensor_op.h index e4198dc4..e39ca9d5 100644 --- a/include/cutlass/epilogue/threadblock/default_thread_map_tensor_op.h +++ b/include/cutlass/epilogue/threadblock/default_thread_map_tensor_op.h @@ -35,7 +35,7 @@ #pragma once -#include "predicated_tile_iterator.h" +#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h" #include "cutlass/gemm/gemm.h" #include "cutlass/layout/pitch_linear.h" diff --git a/include/cutlass/epilogue/threadblock/default_thread_map_volta_tensor_op.h b/include/cutlass/epilogue/threadblock/default_thread_map_volta_tensor_op.h index f0ccd74e..1eac4a18 100644 --- a/include/cutlass/epilogue/threadblock/default_thread_map_volta_tensor_op.h +++ b/include/cutlass/epilogue/threadblock/default_thread_map_volta_tensor_op.h @@ -35,7 +35,7 @@ #pragma once -#include "predicated_tile_iterator.h" +#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h" #include "cutlass/gemm/gemm.h" ///////////////////////////////////////////////////////////////////////////////////////////////// diff --git a/include/cutlass/epilogue/threadblock/default_thread_map_wmma_tensor_op.h b/include/cutlass/epilogue/threadblock/default_thread_map_wmma_tensor_op.h index 9f01d1ff..0dccf652 100644 --- a/include/cutlass/epilogue/threadblock/default_thread_map_wmma_tensor_op.h +++ b/include/cutlass/epilogue/threadblock/default_thread_map_wmma_tensor_op.h @@ -35,7 +35,7 @@ #pragma once -#include "predicated_tile_iterator.h" +#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h" #include "cutlass/gemm/gemm.h" #include "cutlass/layout/pitch_linear.h" diff --git a/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear.h b/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear.h index 1dfde983..1e04c426 100644 --- a/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear.h +++ b/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear.h @@ -45,7 +45,7 @@ #include "cutlass/layout/matrix.h" #include "cutlass/layout/pitch_linear.h" -#include "regular_tile_iterator.h" +#include "cutlass/transform/threadblock/regular_tile_iterator.h" ///////////////////////////////////////////////////////////////////////////////////////////////// diff --git a/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear_2dthreadtile.h b/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear_2dthreadtile.h index a62ff8dd..7fd49598 100644 --- a/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear_2dthreadtile.h +++ b/include/cutlass/transform/threadblock/regular_tile_iterator_pitch_linear_2dthreadtile.h @@ -45,7 +45,7 @@ #include "cutlass/layout/matrix.h" #include "cutlass/layout/pitch_linear.h" -#include "regular_tile_iterator.h" +#include "cutlass/transform/threadblock/regular_tile_iterator.h" /////////////////////////////////////////////////////////////////////////////////////////////////