Unregister updater after test

Change-Id: I8009940bb2ac949f2c3f0d72c943a5b74afd6954
apache · Oct 29, 2020 · a787166 · a787166
1 parent f5924b1
commit a787166
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python/relay/test_external_codegen.py b/tests/python/relay/test_external_codegen.py
@@ -249,6 +249,8 @@ def constant_updater(expr, symbol):
         assert len(params) == 1
         assert "ccompiler_0_p0" in params.keys()
 
+    tvm._ffi.registry.remove_global_func("relay.ext.ccompiler.constant_updater")
+
 
 def test_extern_dnnl():
     if not tvm.get_global_func("relay.ext.dnnl", True):