[llvm] [AMDGPU] Add scaffolding for ML focused scheduling strategy (PR #169616)

Shilei Tian via llvm-commits llvm-commits at lists.llvm.org
Tue Dec 2 11:09:52 PST 2025


================
@@ -331,6 +331,14 @@ void GCNSubtarget::overrideSchedPolicy(MachineSchedPolicy &Policy,
   Policy.OnlyTopDown = false;
   Policy.OnlyBottomUp = false;
 
+  const Function &F = Region.RegionBegin->getMF()->getFunction();
+  Attribute WorkloadAttr = F.getFnAttribute("amdgpu-workload-type");
+  bool IsMLWorkload =
+      WorkloadAttr.isValid() && WorkloadAttr.getValueAsString() == "ml";
+  // Always schedule top-down for better blancing of HW resource usage.
+  if (IsMLWorkload)
----------------
shiltian wrote:

How is this gonna interact with front end?

https://github.com/llvm/llvm-project/pull/169616


More information about the llvm-commits mailing list