Commit d6d9825
ssjia
Update on "[ET-VK] Implement missing Vulkan operators for Parakeet TDT model"
Add missing operators needed for Parakeet TDT model support:
- New symint ops: sym_sub, sym_floordiv, sym_mul in SymIntOps.cpp;
register operator.floordiv and operator.mul as ephemeral ops in
op_registry.py
- New tensor ops: bitwise_not (via unary_op shader with uint8 DTYPE),
logical_and (alias for bitwise_and dispatch)
- Improve _to_copy: expand dtype support to FP_INT_BOOL_T and use
pick_io_storage_fn to restrict to CONTIGUOUS_BUFFER for non-fp
conversions
- Fix where resize: compute output shape via broadcast across all tensor
inputs instead of always using the second input's shape
- Add symint support to split: use extract_int_or_symint_list instead of
get_int_list in resize_split_node and split_with_sizes_copy_default
- Mark scalar_tensor as supporting resize
Differential Revision: [D95970159](https://our.internmc.facebook.com/intern/diff/D95970159/)
cc manuelcandales digantdesai cbilgin
[ghstack-poisoned]2 files changed
Lines changed: 8 additions & 2 deletions
Lines changed: 2 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
45 | 45 | | |
46 | 46 | | |
47 | 47 | | |
48 | | - | |
| 48 | + | |
49 | 49 | | |
50 | | - | |
| 50 | + | |
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
28 | 28 | | |
29 | 29 | | |
30 | 30 | | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
31 | 37 | | |
32 | 38 | | |
33 | 39 | | |
| |||
0 commit comments