optimize

microsoft · Aug 9, 2024 · bcffc49 · bcffc49
1 parent 23f88b7
commit bcffc49
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/bitblas/base/utils.py b/bitblas/base/utils.py
@@ -238,11 +238,12 @@ def tvm_callback_cuda_postproc(code, _):
             continue
         elif map_result.status == StatusKind.COMPLETE:
             idx, code, artifact_path = map_result.value
-            if artifact_path is None:
-                logger.debug("Artifact path is None")
-                continue
             sch = _sched[idx]
             config = configs[idx]
+            if artifact_path is None:
+                ARTIFACT_NOT_FOUND = f"Apply config {config} failed, artifact path is None"
+                logger.debug(ARTIFACT_NOT_FOUND)
+                continue
             rt_mod = tvm.runtime.load_module(artifact_path)
             cpresult = CompileResult(config, sch, rt_mod)
             timer_cuda_mod = rt_mod.time_evaluator(