refactor: Use select instead of branch for softplus non-vec

gabe-l-hart · gabe-l-hart · commit 338acb32b4bb · 2025-12-04T09:28:29.000-07:00
Branch: ggml-cumsum-tri

Signed-off-by: Gabe Goodhart &lt;ghart@us.ibm.com&gt;
diff --git a/ggml/src/ggml-metal/ggml-metal.metal b/ggml/src/ggml-metal/ggml-metal.metal
@@ -1616,7 +1616,7 @@ kernel void kernel_softplus_f32(
         device       float * dst,
         uint tpig[[thread_position_in_grid]]) {
     device const float & x = src0[tpig];
-    dst[tpig] = (x > 20.0f) ? x : log(1.0f + exp(x));
+    dst[tpig] = select(log(1.0f + exp(x)), x, x > 20.0f);
 }
 
 kernel void kernel_softplus_f32_4(

Original file line number	Diff line number	Diff line change
`@@ -1616,7 +1616,7 @@ kernel void kernel_softplus_f32(`
`1616`	`1616`	`device float * dst,`
`1617`	`1617`	`uint tpig[[thread_position_in_grid]]) {`
`1618`	`1618`	`device const float & x = src0[tpig];`
`1619`		`- dst[tpig] = (x > 20.0f) ? x : log(1.0f + exp(x));`
	`1619`	`+ dst[tpig] = select(log(1.0f + exp(x)), x, x > 20.0f);`
`1620`	`1620`	`}`
`1621`	`1621`
`1622`	`1622`	`kernel void kernel_softplus_f32_4(`