Disable loop unrolling in clang

Signed-off-by: Szymon Karpiński <[email protected]>
NVIDIA · Dec 14, 2021 · 58ea73d · 58ea73d
1 parent e972a93
commit 58ea73d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dali/kernels/slice/slice_gpu.cuh b/dali/kernels/slice/slice_gpu.cuh
@@ -139,7 +139,9 @@ __device__ void SliceFunc(OutputType *__restrict__ out, const InputType *__restr
   }
 
   for (; offset < block_end; offset += blockDim.x * coalesced_values<OutputType>) {
+    #ifndef __clang__
     #pragma unroll
+    #endif
     for (uint64_t i = 0; i < coalesced_values<OutputType>; i++) {
       uint64_t idx = offset + i;
       if (idx >= block_end) break;