Support matrix negation in metal backend. (#5891)

shader-slang · Dec 17, 2024 · 9c9e1f7 · 9c9e1f7
1 parent 0a6ffee
commit 9c9e1f7
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/source/slang/slang-emit-metal.cpp b/source/slang/slang-emit-metal.cpp
@@ -701,6 +701,21 @@ bool MetalSourceEmitter::tryEmitInstExprImpl(IRInst* inst, const EmitOpInfo& inO
             m_writer->emit(")");
             return true;
         }
+    case kIROp_Neg:
+        {
+            if (as<IRMatrixType>(inst->getOperand(0)->getDataType()))
+            {
+                // Metal does not support negate operator on matrices,
+                // we should emit "(matrix(0) - op0)" instead.
+                m_writer->emit("(");
+                emitType(inst->getDataType());
+                m_writer->emit("(0) - ");
+                emitOperand(inst->getOperand(0), getInfo(EmitOp::General));
+                m_writer->emit(")");
+                return true;
+            }
+            break;
+        }
     case kIROp_Mul:
         {
             // Component-wise multiplication needs to be special cased,

diff --git a/tests/metal/matrix-negate.slang b/tests/metal/matrix-negate.slang
@@ -0,0 +1,11 @@
+//TEST:SIMPLE(filecheck=CHECK): -target metal
+
+// CHECK: (matrix<float,int(4),int(4)>{{.*}}(0) -
+
+RWStructuredBuffer<float4> output;
+
+[numthreads(1,1,1)]
+void computeMain(uniform float4x4 m)
+{
+    output[0] = (-m)[0];
+}