From 30f6a302283916605ea8cfc6d94501abc26363df Mon Sep 17 00:00:00 2001 From: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com> Date: Wed, 1 Mar 2023 14:54:37 -0600 Subject: [PATCH] Fix tch build issue from the root (#180) (#189) Moved .cargo/config.toml to the root so that the project can be built from the root and burn-tch as well. --- .gitignore | 2 +- scripts/fix-tch-build-arm64.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index 79cf6399f..0f3be7ae4 100644 --- a/.gitignore +++ b/.gitignore @@ -3,4 +3,4 @@ Cargo.lock # These are backup files generated by rustfmt **/*.rs.bk .DS_Store -burn-tch/.cargo/config.toml +.cargo/config.toml diff --git a/scripts/fix-tch-build-arm64.py b/scripts/fix-tch-build-arm64.py index a34d3717b..2c9fba998 100755 --- a/scripts/fix-tch-build-arm64.py +++ b/scripts/fix-tch-build-arm64.py @@ -8,7 +8,7 @@ It's a temporary workaround for https://github.com/burn-rs/burn/issues/180 till tch-rs starts using Torch 2.0 libraries. This script installs torch via pip3 and creates environment variables in -burn-tch/.cargo/config.toml for tch-rs to link cc libs properly. +.cargo/config.toml for tch-rs to link cc libs properly. """ @@ -27,7 +27,7 @@ def update_toml_config(): import tomli_w cargo_cfg_dir = pathlib.Path(__file__).parent.parent.joinpath( - "burn-tch/.cargo").resolve() + ".cargo").resolve() cargo_cfg_dir.exists() if not cargo_cfg_dir.exists(): os.makedirs(cargo_cfg_dir)