diff --git a/src/asahi/compiler/agx_optimizer.c b/src/asahi/compiler/agx_optimizer.c
index e71d579035a..4373a68cff8 100644
--- a/src/asahi/compiler/agx_optimizer.c
+++ b/src/asahi/compiler/agx_optimizer.c
@@ -137,7 +137,8 @@ agx_optimizer_inline_imm(agx_instr **defs, agx_instr *I, unsigned srcs,
            I->op == AGX_OPCODE_LOCAL_ATOMIC) &&
           s != 2)
          continue;
-      if (I->op == AGX_OPCODE_LOCAL_LOAD && s != 1)
+      if ((I->op == AGX_OPCODE_LOCAL_LOAD || I->op == AGX_OPCODE_DEVICE_LOAD) &&
+          s != 1)
          continue;
 
       if (float_src) {