diff --git a/codegen/fix_includes.sh b/codegen/fix_includes.sh index 30379de07e3f..9256fa2a9b8b 100755 --- a/codegen/fix_includes.sh +++ b/codegen/fix_includes.sh @@ -6,9 +6,7 @@ sed -i 's##"torch_xla/csrc/tensor.h"#' $@ # remove the runfiles-prefix used in codegen for pytorch # `torchgen` generates relative includes path and does not support customizing the root, # so we have to fix them up. -sed -i 's#bazel-out/k8-opt/bin/codegen/lazy_tensor_generator.runfiles/torch/##' $@ -sed -i 's#bazel-out/k8-dbg/bin/codegen/lazy_tensor_generator.runfiles/torch/##' $@ -sed -i 's#bazel-out/k8-fastbuild/bin/codegen/lazy_tensor_generator.runfiles/torch/##' $@ +sed -i 's#bazel-out/k8-[^/]*/bin/codegen/lazy_tensor_generator.runfiles/torch/##' $@ # use the generated files that are in the compilation unit sed -i 's##"bazel-out/\1"#' $@