revert the change of nv hpp

codeplaysoftware · Jul 17, 2024 · 3496593 · 3496593
1 parent 6bdda75
commit 3496593
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/include/cutlass/relatively_equal.h b/include/cutlass/relatively_equal.h
@@ -71,7 +71,7 @@ bool relatively_equal_float(T a, T b, T epsilon, T nonzero_floor) {
     return true;
   }
   else if (a == zero || b == zero || diff < nonzero_floor) {
-    return diff < (epsilon * nonzero_floor);
+    return diff < epsilon * nonzero_floor;
   }
 
   return diff < epsilon * (abs_A + abs_B);

diff --git a/tools/util/include/cutlass/util/reference/device/tensor_compare.h b/tools/util/include/cutlass/util/reference/device/tensor_compare.h
@@ -95,18 +95,15 @@ __global__ void
 
   size_t idx = ThreadIdxX() + BlockDimX() * BlockIdxX();
 
-  //for (; idx < capacity; idx += GridDimX() * BlockDimX()) {
-  if (idx < capacity ){
+  for (; idx < capacity; idx += GridDimX() * BlockDimX()) {
     Element a = cutlass::ReferenceFactory<Element>::get(ptr_A, idx);
     Element b = cutlass::ReferenceFactory<Element>::get(ptr_B, idx);
 
     if (!relatively_equal(a, b, epsilon, nonzero_floor)) {
       *equal = 0;
-      //printf("error, idx at: %lu, capacity: %lu, a: %f, b: %f\n", idx, capacity, a, b);
       return;
     }
   }
- // }
 }
 
 } // namespace kernel