Skip to content

Commit 7c23d9e

Browse files
committed
gpu: jit: codegen: handle 64-bit reg/imm addition
1 parent ff209f9 commit 7c23d9e

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/gpu/jit/codegen/kernel.hpp

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,14 @@ class ir_kernel_t : public jit_generator<hw> {
390390
if (src1.is_reg_data()) {
391391
eadd(mod, dst.reg_data(), src0.reg_data(), src1.reg_data());
392392
} else {
393-
eadd(mod, dst.reg_data(), src0.reg_data(), src1.immediate());
393+
if (ngen_is_qw(src1.type())) {
394+
auto tmp = ra_.alloc_sub(src1.type());
395+
emov(1, tmp, src1.immediate());
396+
eadd(mod, dst.reg_data(), src0.reg_data(), tmp);
397+
ra_.safeRelease(tmp);
398+
} else {
399+
eadd(mod, dst.reg_data(), src0.reg_data(), src1.immediate());
400+
}
394401
}
395402
}
396403

0 commit comments

Comments
 (0)