Skip to content

特性支持:xpu_graph支持原位操作 #251

Description

@jyjyjyjyjyjyjyj

Describe the feature🚀

适配的tmo算子:addnorm和bmm 提供了原位但xpu_graph 开启cpp_wrapper时不支持报错。

Image

Image

Image

当前的处理办法:
bmm:tmo提供了非原位操作
addnorm: Model A进了2个,其他网络没进,遇到的场景少。在使用cpp wrapper的时候挂掉了,现在是return False临时绕过。

References

Environment details

XPU_GRAPH: master
model: Model A
mode:
xpu_compiler_backend = xpu_graph.mlu_compiler(
is_training=False,
opt_level=OptLevel.level2,
vendor_compiler_config={"mode":"default", "cpp_wrapper": True},
)

Metadata

Metadata

Labels

enhancementNew feature or request

Type

No type

Fields

No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions