translate -lib=mkldnn to -libs=dnnl in target

apache · Jun 10, 2022 · c8ba892 · c8ba892
1 parent 095ab3e
commit c8ba892
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/tvm/target/target.py b/python/tvm/target/target.py
@@ -108,6 +108,12 @@ def __init__(self, target, host=None):
             When using a dictionary or json string to configure target, the possible values are
             same as target.
         """
+        if "-libs=mkldnn" in target:
+            target = target.replace("mkldnn", "dnnl")
+            warnings.warn(
+                "legacy supoort of mkldnn will be eprecated in the next release."
+                " Please replace -libs=mkldnn to -libs=dnnl to enable Intel OneDNN.",
+            )
         if isinstance(target, (dict, str)):
             target = convert(target)
         if isinstance(host, (dict, str)):