Updates and Bug fixes to CUTLASS 3.3 (#1232)

2023-12-05 09:50:49 -05:00
parent 4a1709e17e
commit e9e30c2304
31 changed files with 534 additions and 227 deletions
@@ -231,7 +231,7 @@ int ceil_div(int a, int b) {
 * log2_up/down codes?
 */
 template <typename value_t>
-CUTLASS_HOST_DEVICE int clz(value_t x) {
+CUTLASS_HOST_DEVICE value_t clz(value_t x) {
  for (int i = 31; i >= 0; --i) {
    if ((1 << i) & x)
      return value_t(31 - i);