Skip to content

Commit fc45eed

Browse files
committed
Update tokenizer path in README for parakeet model (#16495)
1 parent 55fe42b commit fc45eed

File tree

7 files changed

+643
-46
lines changed

7 files changed

+643
-46
lines changed

CMakePresets.json

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,9 @@
183183
],
184184
"cacheVariables": {
185185
"CMAKE_BUILD_TYPE": "Debug",
186-
"CMAKE_INSTALL_PREFIX": "${sourceDir}/cmake-out"
186+
"CMAKE_INSTALL_PREFIX": "${sourceDir}/cmake-out",
187+
"EXECUTORCH_ENABLE_LOGGING": "ON",
188+
"ET_MIN_LOG_LEVEL": "Info"
187189
}
188190
},
189191
{

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)