From 4c0d6e1eb4075401d85df3735248956e512aadfa Mon Sep 17 00:00:00 2001 From: Fujun Han Date: Mon, 25 Apr 2022 04:28:32 +0800 Subject: [PATCH] [BUGFIX]: Force unroll a loop that doesn't have compilation constant (#441) loop times is dangerous. Signed-off-by: Peter Han --- tools/util/include/cutlass/util/device_dump.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/util/include/cutlass/util/device_dump.h b/tools/util/include/cutlass/util/device_dump.h index 6fc8e741..34a113e9 100644 --- a/tools/util/include/cutlass/util/device_dump.h +++ b/tools/util/include/cutlass/util/device_dump.h @@ -101,7 +101,7 @@ CUTLASS_DEVICE void dump_fragment(Fragment const& frag, int N = 0, int M = 0, for (int tid = 0; tid < N; ++tid) { if (tid == thread_id) { printf("TB%d W%d T%d: ", block_id, tid / 32, tid & 31); - CUTLASS_PRAGMA_UNROLL + CUTLASS_PRAGMA_NO_UNROLL for (int i = 0; i < M; i += S) { printf("%.0f ", float(typename Fragment::value_type(frag[i]))); }