File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -344,7 +344,7 @@ def patch_build_sh(path: Path) -> None:
344344rule_start = text.find(rule_name)
345345if rule_start == -1:
346346 raise SystemExit(f"tensor_handle_data rule not found in {path}")
347- rule_end = text.find('\n ) \n ', rule_start)
347+ rule_end = text.find('\\ n) \ \ n', rule_start)
348348if rule_end == -1:
349349 raise SystemExit(f"tensor_handle_data rule end not found in {path}")
350350rule_end += 3
@@ -373,7 +373,7 @@ def patch_build_sh(path: Path) -> None:
373373
374374path = Path(os.environ["TENSOR_HANDLE_DATA_CC"])
375375text = path.read_text(encoding="utf-8")
376- text = text.replace('#include "tensorflow/core/profiler/lib/traceme.h"\n ', '', 1)
376+ text = text.replace('#include "tensorflow/core/profiler/lib/traceme.h"\\ n', '', 1)
377377old = ''' profiler::TraceMe activity(
378378 [caller] { return absl::StrCat(caller, " WaitReady"); },
379379
You can’t perform that action at this time.
0 commit comments