fix pow with scalar input (#170)

ml-explore · Dec 3, 2024 · 30eecff · 30eecff
1 parent c30d558
commit 30eecff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Source/MLX/Ops+Array.swift b/Source/MLX/Ops+Array.swift
@@ -1157,7 +1157,7 @@ public func pow<T: ScalarOrArray>(_ array: MLXArray, _ other: T, stream: StreamO
 public func pow<T: ScalarOrArray>(_ array: T, _ other: MLXArray, stream: StreamOrDevice = .default)
     -> MLXArray
 {
-    let array = other.asMLXArray(dtype: other.dtype)
+    let array = array.asMLXArray(dtype: other.dtype)
     return pow(array, other, stream: stream)
 }