Skip to content

Commit f39750d

Browse files
committed
Update tokenizer path in README for parakeet model (#16495)
1 parent 8e8d97e commit f39750d

File tree

6 files changed

+668
-211
lines changed

6 files changed

+668
-211
lines changed

backends/apple/metal/metal_backend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def get_device_name(cls) -> str:
3232
def get_supported_fallback_kernels(cls) -> Dict[str, Any]:
3333
return {
3434
"aoti_torch_mps_addmm_out": None,
35+
"aoti_torch_mps_bmm_out": None,
3536
"aoti_torch_mps_convolution": None,
3637
"aoti_torch_mps_mm_out": None,
3738
"at::_ops::_scaled_dot_product_attention_math_for_mps::call": None,

backends/apple/metal/runtime/shims/et_metal_ops.h

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,28 @@ AOTITorchError aoti_torch_mps_mm_out(
2727
AOTITensorHandle self,
2828
AOTITensorHandle mat2);
2929

30+
/**
31+
* ExecutorTorch implementation of aoti_torch_mps_addmm_out.
32+
* Performs: out = beta * input + alpha * (mat1 @ mat2)
33+
*/
34+
AOTITorchError aoti_torch_mps_addmm_out(
35+
AOTITensorHandle out,
36+
AOTITensorHandle input,
37+
AOTITensorHandle mat1,
38+
AOTITensorHandle mat2,
39+
int64_t beta,
40+
int64_t alpha);
41+
42+
/**
43+
* ExecutorTorch implementation of aoti_torch_mps_bmm_out.
44+
* Performs batched matrix multiplication: out = self @ mat2
45+
* All tensors must be 3-D with matching batch dimensions.
46+
*/
47+
AOTITorchError aoti_torch_mps_bmm_out(
48+
AOTITensorHandle out,
49+
AOTITensorHandle self,
50+
AOTITensorHandle mat2);
51+
3052
/**
3153
* ExecutorTorch implementation of aoti_torch_mps_convolution.
3254
* Performs 2D convolution operation - matches PyTorch AOTI signature

0 commit comments

Comments
 (0)