Skip to content

GitLab

Unverified Commit c67f73b0 authored by Lei Wang's avatar Lei Wang Committed by GitHub
Browse files

[Env] Optimize the mechanism for locating `TL_LIBS` (#1038)

parent e5399527
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 6 deletions
tilelang/env.py
View file @ c67f73b0
...@@ -4,7 +4,6 @@ import pathlib ...@@ -4,7 +4,6 @@ import pathlib
import logging import logging
import shutil import shutil
import glob import glob
import site
from dataclasses import dataclass from dataclasses import dataclass
from typing import Optional from typing import Optional
...@@ -20,12 +19,9 @@ TL_TEMPLATE_NOT_FOUND_MESSAGE = ("TileLang is not installed or found in the expe ...@@ -20,12 +19,9 @@ TL_TEMPLATE_NOT_FOUND_MESSAGE = ("TileLang is not installed or found in the expe
", which may lead to compilation bugs when utilize tilelang backend." ", which may lead to compilation bugs when utilize tilelang backend."
TVM_LIBRARY_NOT_FOUND_MESSAGE = ("TVM is not installed or found in the expected path") TVM_LIBRARY_NOT_FOUND_MESSAGE = ("TVM is not installed or found in the expected path")
SITE_PACKAGES = site.getsitepackages()
TL_LIBS = [os.path.join(i, 'tilelang/lib') for i in site.getsitepackages()]
TL_LIBS = [i for i in TL_LIBS if os.path.exists(i)]
TL_ROOT = os.path.dirname(os.path.abspath(__file__)) TL_ROOT = os.path.dirname(os.path.abspath(__file__))
TL_LIBS = [os.path.join(i, 'lib') for i in [TL_ROOT]]
TL_LIBS = [i for i in TL_LIBS if os.path.exists(i)]
DEV = False DEV = False
THIRD_PARTY_ROOT = os.path.join(TL_ROOT, '3rdparty') THIRD_PARTY_ROOT = os.path.join(TL_ROOT, '3rdparty')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment