From 70d5cc62ca7beb5f9f5148de9a779dd213bf6d5e Mon Sep 17 00:00:00 2001
From: Joseph Melber <jgmelber@gmail.com>
Date: Tue, 9 Apr 2024 12:06:00 -0600
Subject: [PATCH] [ASPLOS] Rename directories (#1196)

Co-authored-by: Jeff Fifield <jeff.fifield@amd.com>
Co-authored-by: AndraBisca <andrab@amd.com>
Co-authored-by: Jack Lo <36210336+jackl-xilinx@users.noreply.github.com>
---
 mlir_tutorials/lit.cfg.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mlir_tutorials/lit.cfg.py b/mlir_tutorials/lit.cfg.py
index e0f28f17ff..77226c1b62 100755
--- a/mlir_tutorials/lit.cfg.py
+++ b/mlir_tutorials/lit.cfg.py
@@ -7,9 +7,11 @@
 # Copyright (C) 2022, Advanced Micro Devices, Inc.
 
 import os
+import platform
 import re
 import shutil
 import subprocess
+import tempfile
 
 import lit.formats
 import lit.util
@@ -22,6 +24,9 @@
 config.name = "AIE_TUTORIALS"
 
 config.test_format = lit.formats.ShTest(not llvm_config.use_lit_shell)
+config.environment["PYTHONPATH"] = "{}".format(
+    os.path.join(config.aie_obj_root, "python")
+)
 
 # suffixes: A list of file extensions to treat as test files.
 config.suffixes = [".mlir"]