[PATCH] D155851: [llvm][nvptx] Add sm_90a

guray ozen via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Thu Jul 20 08:29:06 PDT 2023


guraypp created this revision.
Herald added subscribers: mattd, gchakrabarti, asavonic, hiraditya.
Herald added a project: All.
guraypp requested review of this revision.
Herald added subscribers: llvm-commits, wangpc, jholewinski.
Herald added a project: LLVM.

This works adds `sm_90a` as nvptx target. `sm_90a` is required to generate  wgmma and setmaxnreg instructions.

Here is information about "a" prefix in PTX document:
Target architectures with suffix “a”, such as sm_90a, include architecture-accelerated features that are supported on the specified architecture only, hence such targets do not follow the onion layer model. Therefore, PTX code generated for such targets cannot be run on later generation devices. Architecture-accelerated features can only be used with targets that support these features.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D155851

Files:
  llvm/lib/Target/NVPTX/NVPTX.td


Index: llvm/lib/Target/NVPTX/NVPTX.td
===================================================================
--- llvm/lib/Target/NVPTX/NVPTX.td
+++ llvm/lib/Target/NVPTX/NVPTX.td
@@ -41,6 +41,7 @@
                    70, 71, 72, 73, 74, 75, 76, 77, 78, 80, 81] in
   def PTX#version: FeaturePTX<version>;
 
+def SM90a: FeatureSM<90>;
 //===----------------------------------------------------------------------===//
 // NVPTX supported processors.
 //===----------------------------------------------------------------------===//
@@ -68,6 +69,7 @@
 def : Proc<"sm_87", [SM87, PTX74]>;
 def : Proc<"sm_89", [SM89, PTX78]>;
 def : Proc<"sm_90", [SM90, PTX78]>;
+def : Proc<"sm_90a", [SM90a, PTX80]>;
 
 def NVPTXInstrInfo : InstrInfo {
 }


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D155851.542545.patch
Type: text/x-patch
Size: 734 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20230720/92092c18/attachment.bin>


More information about the llvm-commits mailing list