gpu: jit: codegen: handle 64-bit reg/imm addition

oneapi-src · May 12, 2023 · 7c23d9e · 7c23d9e
1 parent ff209f9
commit 7c23d9e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/gpu/jit/codegen/kernel.hpp b/src/gpu/jit/codegen/kernel.hpp
@@ -390,7 +390,14 @@ class ir_kernel_t : public jit_generator<hw> {
         if (src1.is_reg_data()) {
             eadd(mod, dst.reg_data(), src0.reg_data(), src1.reg_data());
         } else {
-            eadd(mod, dst.reg_data(), src0.reg_data(), src1.immediate());
+            if (ngen_is_qw(src1.type())) {
+                auto tmp = ra_.alloc_sub(src1.type());
+                emov(1, tmp, src1.immediate());
+                eadd(mod, dst.reg_data(), src0.reg_data(), tmp);
+                ra_.safeRelease(tmp);
+            } else {
+                eadd(mod, dst.reg_data(), src0.reg_data(), src1.immediate());
+            }
         }
     }