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(
1474
1474
private func bicubicInterpolate(
1475
1475
_ x: MLXArray , to targetSize: ( Int , Int ) , alignCorners: Bool = false
1476
1476
) -> 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
1478
1478
// 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
+ //
1483
1481
// Input: NHWC format [batch, height, width, channels]
1484
1482
// Output: NHWC format [batch, target_height, target_width, channels]
1485
1483
You can’t perform that action at this time.
0 commit comments