已合并
update to support torch-2.15.0 #5719
SCh_zx创建于 3 天前
update to support torch-2.15.0 #5719
已合并
共 7 个文件变更+20-6
| @@ -21,7 +21,7 @@ CDIR="$(cd "$(dirname "$0")" ; pwd -P)" | |||
| 21 | cd $CDIR/ | 21 | cd $CDIR/ |
| 22 | 22 | ||
| 23 | # Following updates only add incompatible ops files in corresponding branch folder. | 23 | # Following updates only add incompatible ops files in corresponding branch folder. |
| 24 | newest_minor_version=14 | 24 | newest_minor_version=15 |
| 25 | for minor_version in $(seq 1 ${newest_minor_version}); do | 25 | for minor_version in $(seq 1 ${newest_minor_version}); do |
| 26 | # Merge base info and version related info (unsupported ops) | 26 | # Merge base info and version related info (unsupported ops) |
| 27 | sed -i "1r test/unsupported_ops_info_base.yaml" test/test_v2r${minor_version}_ops/unsupported_ops_info.yaml | 27 | sed -i "1r test/unsupported_ops_info_base.yaml" test/test_v2r${minor_version}_ops/unsupported_ops_info.yaml |
| @@ -1,4 +1,4 @@ | |||
| 1 | all_version: [v2.1, v2.2, v2.3, v2.4, v2.5, v2.6, v2.7, v2.8, v2.9, v2.10, v2.11, v2.12, v2.13, v2.14] | 1 | all_version: [v2.1, v2.2, v2.3, v2.4, v2.5, v2.6, v2.7, v2.8, v2.9, v2.10, v2.11, v2.12, v2.13, v2.14, v2.15] |
| 2 | backward: | 2 | backward: |
| 3 | - name: linalg_cholesky(Tensor self, *, bool upper=False) -> Tensor | 3 | - name: linalg_cholesky(Tensor self, *, bool upper=False) -> Tensor |
| 4 | self: cholesky_backward(grad, upper, result) | 4 | self: cholesky_backward(grad, upper, result) |
| @@ -1,4 +1,4 @@ | |||
| 1 | all_version: [v2.1, v2.2, v2.3, v2.4, v2.5, v2.6, v2.7, v2.8, v2.9, v2.10, v2.11, v2.12, v2.13, v2.14] | 1 | all_version: [v2.1, v2.2, v2.3, v2.4, v2.5, v2.6, v2.7, v2.8, v2.9, v2.10, v2.11, v2.12, v2.13, v2.14, v2.15] |
| 2 | official: | 2 | official: |
| 3 | - func: __ilshift__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) | 3 | - func: __ilshift__.Scalar(Tensor(a!) self, Scalar other) -> Tensor(a!) |
| 4 | acl_op: all_version | 4 | acl_op: all_version |
| @@ -28,5 +28,6 @@ | |||
| 28 | 28 | ||
| 29 | 29 | ||
| 30 | 30 | ||
| 31 | #define VERSION_NEWEST V2R14 | 31 | #define V2R15 14 |
| 32 | |||
| 32 | 33 | ||
| @@ -117,6 +117,19 @@ | |||
| 117 | "npu_ring_attention_update", | 117 | "npu_ring_attention_update", |
| 118 | "npu_renorm_backward" | 118 | "npu_renorm_backward" |
| 119 | ], | 119 | ], |
| 120 | "v2.15": [ | ||
| 121 | "npu_gelu_mul", | ||
| 122 | "npu_clipped_swiglu", | ||
| 123 | "npu_sim_exponential_", | ||
| 124 | "npu_fused_floyd_attention", | ||
| 125 | "npu_fused_floyd_attention_backward", | ||
| 126 | "npu_fused_causal_conv1d", | ||
| 127 | "npu_masked_causal_conv1d", | ||
| 128 | "npu_masked_causal_conv1d_backward", | ||
| 129 | "npu_apply_rotary_pos_emb", | ||
| 130 | "npu_ring_attention_update", | ||
| 131 | "npu_renorm_backward" | ||
| 132 | ], | ||
| 120 | "all_version": [ | 133 | "all_version": [ |
| 121 | "_npu_dropout", | 134 | "_npu_dropout", |
| 122 | "copy_memory_", | 135 | "copy_memory_", |
| @@ -0,0 +1,2 @@ | |||
| 1 | # base unsupported ops info (fixed, obtained from unsupported_ops_info_base.yaml during code generation) | ||
| 2 | # Add unsupported ops info of 2.15 below | ||
| @@ -39,8 +39,6 @@ F = TypeVar( | |||
| 39 | # ITensorListRef on those versions and we must match. Since IListRef has an | 39 | # ITensorListRef on those versions and we must match. Since IListRef has an |
| 40 | # implicit constructor from ArrayRef, this signature is also accepted from | 40 | # implicit constructor from ArrayRef, this signature is also accepted from |
| 41 | # older wrappers that still pass TensorList -- so no version gating is needed. | 41 | # older wrappers that still pass TensorList -- so no version gating is needed. |
| 42 | # See upstream pytorch#185272 for _scaled_mm_v2 (landed 2026-06-15, releasing | ||
| 43 | # in 2.14). | ||
| 44 | _FORCE_ILISTREF_TENSOR_LIST_OPS = { | 42 | _FORCE_ILISTREF_TENSOR_LIST_OPS = { |
| 45 | "_scaled_mm_v2", | 43 | "_scaled_mm_v2", |
| 46 | "_scaled_grouped_mm_v2", | 44 | "_scaled_grouped_mm_v2", |