fix riscv64 cpuinfo patch

cocoa-xu · Mar 17, 2024 · 3e73d60 · 3e73d60
1 parent c5ffe96
commit 3e73d60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/patches/apply_patch.py b/patches/apply_patch.py
@@ -79,7 +79,7 @@ def patch_cpuinfo_riscv64_sys_hwprobe(tf_version: str, tf_src_root: str):
     with open(cpuinfo_cmake, 'r') as source:
         for line in source:
             line_strip = line.strip()
-            if not patched_1 and line_strip == '# Sync with tensorflow/third_party/cpuinfo/workspace.bzl':
+            if not patched_1 and line_strip == '# Sync with tensorflow/workspace2.bzl':
                 fixed.write(f"  {line_strip} # fixed\n")
                 fixed.write("  GIT_TAG 6543fec09b2f04ac4a666882998b534afc9c1349\n")
                 patched_1 = True