Fix Zephyr flashing on physical hardware, busted in #7813 (#7853)

apache · Apr 15, 2021 · 1ebfafd · 1ebfafd
1 parent 2968c9a
commit 1ebfafd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/tvm/micro/contrib/zephyr.py b/python/tvm/micro/contrib/zephyr.py
@@ -465,9 +465,10 @@ def flash(self, micro_binary):
         if self._qemu:
             return self._zephyr_transport(micro_binary)
 
-        build_dir = os.path.dirname(
-            micro_binary.abspath(micro_binary.labelled_files["cmake_cache"][0])
-        )
+        cmake_cache_path = micro_binary.abspath(micro_binary.labelled_files["cmake_cache"][0])
+        cmake_entries = read_cmake_cache(cmake_cache_path)
+
+        build_dir = os.path.dirname(cmake_cache_path)
 
         # The nRF5340DK requires an additional `nrfjprog --recover` before each flash cycle.
         # This is because readback protection is enabled by default when this device is flashed.