We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0adbe8c commit f0b8e71Copy full SHA for f0b8e71
2 files changed
.github/workflows/mlx.yml
@@ -9,6 +9,7 @@ on:
9
paths:
10
- .github/workflows/mlx.yml
11
- backends/mlx/**
12
+ - extension/llm/export/**
13
workflow_dispatch:
14
15
permissions: {}
backends/mlx/ops.py
@@ -219,12 +219,15 @@ def normalize_reduction_dim(
219
return dim, keepdim
220
221
222
+<<<<<<< HEAD
223
@REGISTRY.register(target=["NOOP", torch.ops.aten._assert_scalar.default])
224
def _noop_handler(P: MLXProgramBuilder, n: Node) -> None:
225
"""No-op handler for nodes that don't emit any MLX instructions."""
226
return None
227
228
229
+=======
230
+>>>>>>> e3b488076f (up)
231
@REGISTRY.register(target=[torch.ops.aten.addmm.default])
232
def _addmm_handler(P: MLXProgramBuilder, n: Node) -> Slot:
233
"""Handle addmm: self + (mat1 @ mat2).
0 commit comments