benchdnn: add size_t overload for rnd_up

Author: rjoursler

tests/benchdnn/common.cpp

@@ -508,11 +508,21 @@ int64_t div_up(const int64_t a, const int64_t b) {
     return (a + b - 1) / b;
 }
 
+size_t div_up(const size_t a, const size_t b) {
+    SAFE_V(b != 0 ? OK : FAIL);
+    return (a + b - 1) / b;
+}
+
 int64_t rnd_up(const int64_t a, const int64_t b) {
     SAFE_V(b != 0 ? OK : FAIL);
     return div_up(a, b) * b;
 }
 
+size_t rnd_up(const size_t a, const size_t b) {
+    SAFE_V(b != 0 ? OK : FAIL);
+    return div_up(a, b) * b;
+}
+
 int64_t next_pow2(int64_t a) {
     assert(a > 0 && a <= ((int64_t)1 << 62));
     if (a > 1) a--;

tests/benchdnn/common.hpp

@@ -175,7 +175,9 @@ int batch(const char *fname, bench_f bench);
 int flip_coin(ptrdiff_t seed, float probability);
 
 int64_t div_up(const int64_t a, const int64_t b);
+size_t div_up(const size_t a, const size_t b);
 int64_t rnd_up(const int64_t a, const int64_t b);
+size_t rnd_up(const size_t a, const size_t b);
 int64_t next_pow2(int64_t a);
 int mxcsr_cvt(float f);