Llm and sd additional ops (#20435)
* TorchFX: New ops added (baddbbmm, leaky_relu_) * TorchFX: Initial scaled_dot_product_flash_attention * Code Formatting: scaled_fot_product_attention translation * TorchFX unit test enabled for SDPA * Typo fix in comment line Co-authored-by: Maxim Vafin <maxim.vafin@intel.com> --------- Co-authored-by: Maxim Vafin <maxim.vafin@intel.com>
This commit is contained in:
1 parent
5018be82c3
commit
3d5fe8d446
4 files changed
+37
-7
No files matched your search
@@ -41,6 +41,7 @@ class OperatorSupport(OperatorSupport):
|
||||
"torch.ops.aten.arange.default": None,
|
||||
"torch.ops.aten.argmax.default": None,
|
||||
"torch.ops.aten.avg_pool2d.default": None,
|
||||
"torch.ops.aten.baddbmm.default": None,
|
||||
"torch.ops.aten.bitwise_and.Tensor": None,
|
||||
"torch.ops.aten.bmm.default": None,
|
||||
"torch.ops.aten.cat.default": None,
|
||||
@@ -67,6 +68,7 @@ class OperatorSupport(OperatorSupport):
|
||||
"torch.ops.aten.hardswish_.default": None,
|
||||
"torch.ops.aten.hardtanh_.default": None,
|
||||
"torch.ops.aten.index.Tensor": None,
|
||||
"torch.ops.aten.leaky_relu_.default": None,
|
||||
"torch.ops.aten.lift_fresh_copy.default": None,
|
||||
"torch.ops.aten.linalg_vector_norm.default": None,
|
||||
"torch.ops.aten.lt.Tensor": None,
|
||||
@@ -89,6 +91,7 @@ class OperatorSupport(OperatorSupport):
|
||||
"torch.ops.aten.relu.default": None,
|
||||
"torch.ops.aten.relu_.default": None,
|
||||
"torch.ops.aten.rsub.Scalar": None,
|
||||
"torch.ops.aten._scaled_dot_product_flash_attention.default": None,
|
||||
"torch.ops.aten.select.int": None,
|
||||
"torch.ops.aten.sigmoid.default": None,
|
||||
"torch.ops.aten.silu.default": None,
|
||||
|
||||
Reference in new issue
Block a user