[llvm] r371407 - AMDGPU/GlobalISel: Legalize wavefrontsize intrinsic

[Matt Arsenault via llvm-commits]

Author: arsenm
Date: Mon Sep  9 08:20:49 2019
New Revision: 371407

URL: http://llvm.org/viewvc/llvm-project?rev=371407&view=rev
Log:
AMDGPU/GlobalISel: Legalize wavefrontsize intrinsic

Added:
    llvm/trunk/test/CodeGen/AMDGPU/GlobalISel/legalize-amdgcn.wavefrontsize.mir
Modified:
    llvm/trunk/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp

Modified: llvm/trunk/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp?rev=371407&r1=371406&r2=371407&view=diff
==============================================================================
--- llvm/trunk/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp (original)
+++ llvm/trunk/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp Mon Sep  9 08:20:49 2019
@@ -1534,6 +1534,12 @@ bool AMDGPULegalizerInfo::legalizeIntrin
     return legalizeIsAddrSpace(MI, MRI, B, AMDGPUAS::LOCAL_ADDRESS);
   case Intrinsic::amdgcn_is_private:
     return legalizeIsAddrSpace(MI, MRI, B, AMDGPUAS::PRIVATE_ADDRESS);
+  case Intrinsic::amdgcn_wavefrontsize: {
+    B.setInstr(MI);
+    B.buildConstant(MI.getOperand(0), ST.getWavefrontSize());
+    MI.eraseFromParent();
+    return true;
+  }
   default:
     return true;
   }

Added: llvm/trunk/test/CodeGen/AMDGPU/GlobalISel/legalize-amdgcn.wavefrontsize.mir
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/AMDGPU/GlobalISel/legalize-amdgcn.wavefrontsize.mir?rev=371407&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/AMDGPU/GlobalISel/legalize-amdgcn.wavefrontsize.mir (added)
+++ llvm/trunk/test/CodeGen/AMDGPU/GlobalISel/legalize-amdgcn.wavefrontsize.mir Mon Sep  9 08:20:49 2019
@@ -0,0 +1,18 @@
+# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
+# RUN: llc -mtriple=amdgcn-mesa-mesa3d -mcpu=tahiti -run-pass=legalizer %s -o - | FileCheck -check-prefix=WAVE64 %s
+# RUN: llc -mtriple=amdgcn-mesa-mesa3d -mcpu=gfx1010 -run-pass=legalizer %s -o - | FileCheck -check-prefix=WAVE32 %s
+
+---
+name: test_wavefrontsize
+body: |
+  bb.0:
+
+    ; WAVE64-LABEL: name: test_wavefrontsize
+    ; WAVE64: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 64
+    ; WAVE64: $vgpr0 = COPY [[C]](s32)
+    ; WAVE32-LABEL: name: test_wavefrontsize
+    ; WAVE32: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 32
+    ; WAVE32: $vgpr0 = COPY [[C]](s32)
+    %0:_(s32) = G_INTRINSIC intrinsic(@llvm.amdgcn.wavefrontsize)
+    $vgpr0 = COPY %0
+...