youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more robust imports in heuristics.py and heuristics_provider.py (#2596)

Browse Source
This commit is contained in:
Harrison Barclay
2025-08-28 22:32:55 -04:00
committed by GitHub
parent 496654bf2c
commit b2dd65dc86
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/cutlass_library/heuristics.py
View File

@ -37,7 +37,8 @@ import json
import csv
try:
if CUTLASS_IGNORE_PACKAGE:
import builtins
if hasattr(builtins, "CUTLASS_IGNORE_PACKAGE") and CUTLASS_IGNORE_PACKAGE == True:
raise ImportError("Disabling attempt to import cutlass_library")
from cutlass_library.library import *
from cutlass_library.generator import *

8
python/cutlass_library/heuristics_provider.py
View File

@ -41,7 +41,13 @@ import logging
import ctypes
import functools
from library import DataType, LayoutType
try:
import builtins
if hasattr(builtins, "CUTLASS_IGNORE_PACKAGE") and CUTLASS_IGNORE_PACKAGE == True:
raise ImportError("Disabling attempt to import cutlass_library")
from cutlass_library.library import DataType, LayoutType
except ImportError:
from library import DataType, LayoutType
class MatmulHeuristics: