Files
rocm/patches/tensile-5.5.patch
2024-09-20 14:35:41 -07:00

26 lines
1.1 KiB
Diff

diff --git a/Tensile/TensileCreateLibrary.py b/Tensile/TensileCreateLibrary.py
index 308d61fd..4037ec7c 100644
--- a/Tensile/TensileCreateLibrary.py
+++ b/Tensile/TensileCreateLibrary.py
@@ -1078,13 +1078,18 @@ def generateLogicDataAndSolutions(logicFiles, args):
# logicData[problemType] = []
# logicData[problemType].append((scheduleName, deviceNames, \
# solutionsForSchedule, indexOrder, exactLogic, rangeLogic ))
-
+ (archs, _) = splitArchs()
if globalParameters["SeparateArchitectures"] or globalParameters["LazyLibraryLoading"]:
if "fallback" in masterLibraries.keys():
for key, value in masterLibraries.items():
if key != "fallback":
value.merge(deepcopy(masterLibraries["fallback"]))
-
+ for archName in archs:
+ archName = archName.split('-', 1)[0]
+ if archName not in masterLibraries:
+ print1("Using fallback for arch: " + archName)
+ masterLibraries[archName] = deepcopy(masterLibraries["fallback"])
+ masterLibraries[archName].version = args.version
masterLibraries.pop("fallback")
for _, masterLibrary in masterLibraries.items():