feat: tensile rocm 5.5 patch

2024-09-20 14:35:41 -07:00
parent 9bff3fafb1
commit ae6d8bf9ac
2 changed files with 39 additions and 3 deletions
@@ -0,0 +1,25 @@
+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():