File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -1474,12 +1474,10 @@ private func gemma3nAttentionWithCacheUpdate(
14741474private func bicubicInterpolate(
14751475 _ x: MLXArray , to targetSize: ( Int , Int ) , alignCorners: Bool = false
14761476) -> MLXArray {
1477- // TODO: This implementation uses nested loops and sequential MLX operations, which is much slower
1477+ // This implementation uses nested loops and sequential MLX operations, which is much slower
14781478 // than the Python version that uses mx.fast.metal_kernel() for parallel GPU computation.
1479- // MLX Swift currently doesn't have custom Metal kernel creation capabilities like Python's
1480- // mx.fast.metal_kernel(). Consider optimizing with vectorized MLX operations or requesting
1481- // custom kernel support from the MLX Swift team for better performance.
1482-
1479+ // TODO: Port the custom metal kernel from Python to Swift using `MLXFast.metalKernel`.
1480+ //
14831481 // Input: NHWC format [batch, height, width, channels]
14841482 // Output: NHWC format [batch, target_height, target_width, channels]
14851483
You can’t perform that action at this time.
0 commit comments