fix overflow for workspace size (#1446)

18234a58 · Umang Yadav · GitHub · 1820198e · 18234a58
Unverified Commit 18234a58 authored Nov 06, 2022 by Umang Yadav Committed by GitHub Nov 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/compile_miopen.cpp src/targets/gpu/compile_miopen.cpp +1 -1

No files found.
--- a/src/targets/gpu/compile_miopen.cpp
+++ b/src/targets/gpu/compile_miopen.cpp
@@ -64,7 +64,7 @@ std::size_t compile_miopen::compile(operation& op, instruction_ref ins, bool for
 {
    op.from_value({{"int8_x4_format", format}});
    auto v = op.compile(*ctx, ins->get_shape(), to_shapes(ins->inputs()));
-    return v.get("workspace", 0);
+    return v.get<std::size_t>("workspace", 0);
 }
 void compile_miopen::apply(module& m) const