Remove the need to use vector

ROCm · Dec 10, 2024 · 86d9ee3 · 86d9ee3
1 parent 005d8c7
commit 86d9ee3
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/onnx/parse_matmulnbits.cpp b/src/onnx/parse_matmulnbits.cpp
@@ -66,11 +66,10 @@ struct parse_matmulnbits : op_parser<parse_matmulnbits>
                            to_string_range(expected_b_lens) +
                            ". Actual dims: " + to_string_range(args[1]->get_shape().lens()));
 
-        std::vector<size_t> expected_scales_lens{n * n_blocks_per_col};
-
-        if(args[2]->get_shape().elements() != expected_scales_lens[0])
+        const size_t expected_scales_lens = n * n_blocks_per_col;
+        if(args[2]->get_shape().elements() != expected_scales_lens)
             MIGRAPHX_THROW("MatMulNBits: Input scales does not match expected dims: " +
-                           to_string_range(expected_scales_lens) +
+                           to_string(expected_scales_lens) +
                            ". Actual dims: " + to_string_range(args[2]->get_shape().lens()));
 
         if(args.size() > 3)