commit e8ade89bb3803fe07534b20db6b35bbc7ffc42e3
author Valery Pykhtin <Valery.Pykhtin@amd.com> Fri Sep 06 15:33:53 2019 +0000
committer Valery Pykhtin <Valery.Pykhtin@amd.com> Fri Sep 06 15:33:53 2019 +0000
tree 23989d641e6d919e80ea97c618f804145fb9049f
parent 8f10a5f42d5b391efe57a982c98de5e1ae245e4d

[AMDGPU] Enable constant offset promotion to immediate operand for VMEM stores

Differential revision: https://reviews.llvm.org/D66958

llvm-svn: 371214

llvm/lib/Target/AMDGPU/SILoadStoreOptimizer.cpp

diff --git a/llvm/lib/Target/AMDGPU/SILoadStoreOptimizer.cpp b/llvm/lib/Target/AMDGPU/SILoadStoreOptimizer.cpp
index 70fb377..49f3481 100644
--- a/llvm/lib/Target/AMDGPU/SILoadStoreOptimizer.cpp
+++ b/llvm/lib/Target/AMDGPU/SILoadStoreOptimizer.cpp
@@ -1316,13 +1316,14 @@
     MemInfoMap &Visited,
     SmallPtrSet<MachineInstr *, 4> &AnchorList) {
 
-  // TODO: Support flat and scratch.
-  if (AMDGPU::getGlobalSaddrOp(MI.getOpcode()) < 0 ||
-      TII->getNamedOperand(MI, AMDGPU::OpName::vdata) != NULL)
+  if (!(MI.mayLoad() ^ MI.mayStore()))
     return false;
 
-  // TODO: Support Store.
-  if (!MI.mayLoad())
+  // TODO: Support flat and scratch.
+  if (AMDGPU::getGlobalSaddrOp(MI.getOpcode()) < 0)
+    return false;
+
+  if (MI.mayLoad() && TII->getNamedOperand(MI, AMDGPU::OpName::vdata) != NULL)
     return false;
 
   if (AnchorList.count(&MI))