未验证 提交 2a01a157 编写于 作者: H huzhiqiang 提交者: GitHub

[Infrt] skip grad kernel in infrt frame (#41315)

* code

* code
上级 50714d5c
...@@ -19,6 +19,13 @@ import json ...@@ -19,6 +19,13 @@ import json
skip_list = ["adam_sig.cc", "adamw_sig.cc"] skip_list = ["adam_sig.cc", "adamw_sig.cc"]
def is_grad_kernel(kernel_info):
kernel_name = kernel_info.split(",")[0]
if kernel_name.endswith("_grad"):
return True
return False
def parse_compat_registry(kernel_info): def parse_compat_registry(kernel_info):
name, inputs_str, attrs_str, outputs_str = kernel_info.split(",{") name, inputs_str, attrs_str, outputs_str = kernel_info.split(",{")
kernel_info = {} kernel_info = {}
...@@ -62,6 +69,8 @@ def get_compat_kernels_info(): ...@@ -62,6 +69,8 @@ def get_compat_kernels_info():
"").strip("return").strip("KernelSignature(").strip( "").strip("return").strip("KernelSignature(").strip(
"\);").replace("\"", "").replace("\\", "") "\);").replace("\"", "").replace("\\", "")
registry = False registry = False
if is_grad_kernel(data):
continue
name, registry_info = parse_compat_registry(data) name, registry_info = parse_compat_registry(data)
if name in kernels_info: if name in kernels_info:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册