Skip to content

Commit

Permalink
remove tools/ttnn-standalone from PATH in llvm-lit; remove prints
Browse files Browse the repository at this point in the history
  • Loading branch information
svuckovicTT committed Jan 9, 2025
1 parent ad714c3 commit 5b4f4bf
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions test/lit.cfg.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,30 +101,18 @@ def set_system_desc_features(system_desc):

# Add `TT_MLIR_HOME` to lit environment.
if "TT_MLIR_HOME" in os.environ:
print(f"{os.environ['TT_MLIR_HOME']}")
llvm_config.with_environment("TT_MLIR_HOME", os.environ["TT_MLIR_HOME"])

# Additionally, add 'tools/ttnn-standalone' to $PATH
llvm_config.with_environment(
"PATH",
[
os.path.join(os.getenv("TT_MLIR_HOME"), "tools/ttnn-standalone"),
],
append_path=True,
)
else:
raise OSError("Error: TT_MLIR_HOME not set")

# Add `TT_METAL_HOME` to lit environment.
if "TT_METAL_HOME" in os.environ:
print(f"{os.environ['TT_METAL_HOME']}")
llvm_config.with_environment("TT_METAL_HOME", os.environ["TT_METAL_HOME"])
else:
raise OSError("Error: TT_METAL_HOME not set")

# Add `TT_METAL_BUILD_HOME` to lit environment.
if "TT_METAL_BUILD_HOME" in os.environ:
print(f"{os.environ['TT_METAL_BUILD_HOME']}")
llvm_config.with_environment(
"TT_METAL_BUILD_HOME", os.environ["TT_METAL_BUILD_HOME"]
)
Expand All @@ -133,7 +121,6 @@ def set_system_desc_features(system_desc):

# Add `ARCH_NAME` to lit environment.
if "ARCH_NAME" in os.environ:
print(f"ARCH_NAME={os.environ['ARCH_NAME']}")
llvm_config.with_environment("ARCH_NAME", os.environ["ARCH_NAME"])
else:
raise OSError("Error: ARCH_NAME not set.")

0 comments on commit 5b4f4bf

Please sign in to comment.