From fc1e5bbb56de9a7f844f16cb2bc67e1252460994 Mon Sep 17 00:00:00 2001
From: "Korovin, Vladislav" <vladislav.korovin@intel.com>
Date: Mon, 23 Oct 2023 17:59:44 +0000
Subject: [PATCH]  Fix addrspacecast folding in GepLowering

.
---
 .../lib/GenXCodeGen/GenXGEPLowering.cpp       |  7 +++---
 .../test/GenXGEPLowering/fold-ptrtoint.ll     | 23 ++++++++++++++++++-
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/IGC/VectorCompiler/lib/GenXCodeGen/GenXGEPLowering.cpp b/IGC/VectorCompiler/lib/GenXCodeGen/GenXGEPLowering.cpp
index 6c0fd41e43bf..6a58bddd5e39 100644
--- a/IGC/VectorCompiler/lib/GenXCodeGen/GenXGEPLowering.cpp
+++ b/IGC/VectorCompiler/lib/GenXCodeGen/GenXGEPLowering.cpp
@@ -16,8 +16,11 @@ SPDX-License-Identifier: MIT
 
 #include "GenX.h"
 #include "GenXModule.h"
+#include "GenXUtil.h"
 #include "Probe/Assertion.h"
 
+#include "vc/Utils/General/Types.h"
+
 #include "llvmWrapper/IR/DerivedTypes.h"
 #include "llvmWrapper/IR/Instructions.h"
 #include "llvmWrapper/Support/TypeSize.h"
@@ -130,9 +133,7 @@ Value *GenXGEPLowering::visitPtrToIntInst(PtrToIntInst &PTI) {
   for (auto *Cast = dyn_cast<CastInst>(Src); Cast != nullptr;
        Cast = dyn_cast<CastInst>(Src)) {
     if (isa<AddrSpaceCastInst>(Cast)) {
-      auto *PtrTy = cast<PointerType>(Cast->getType());
-      auto AddrSpace = PtrTy->getAddressSpace();
-      if (AddrSpace != TTI->getFlatAddressSpace())
+      if (!genx::isNoopCast(Cast))
         break;
       // The `addrspacecast` is just no-op so it can be eliminated
       Src = Cast->getOperand(0);
diff --git a/IGC/VectorCompiler/test/GenXGEPLowering/fold-ptrtoint.ll b/IGC/VectorCompiler/test/GenXGEPLowering/fold-ptrtoint.ll
index ad4dd9183f51..61b5b04a0378 100644
--- a/IGC/VectorCompiler/test/GenXGEPLowering/fold-ptrtoint.ll
+++ b/IGC/VectorCompiler/test/GenXGEPLowering/fold-ptrtoint.ll
@@ -8,8 +8,9 @@
 
 ; RUN: %opt %use_old_pass_manager% -GenXGEPLowering -march=genx64 -mcpu=Gen9 -S < %s | FileCheck %s
 
-target datalayout = "e-p:64:64-p6:32:32-i64:64-n8:16:32:64"
+target datalayout = "e-p:64:64-p3:32:32-i64:64-n8:16:32:64"
 
+declare <32 x double> @llvm.vc.internal.lsc.load.slm.v32f64.v32i1.v32i32(<32 x i1>, i8, i8, i8, i8, i8, i32, <32 x i32>, i16, i32, <32 x double>)
 declare <32 x i32> @llvm.vc.internal.lsc.load.ugm.v32i32.v1i1.i64(<1 x i1>, i8, i8, i8, i8, i8, i64, i64, i16, i32, <32 x i32>)
 declare <32 x double> @llvm.vc.internal.lsc.load.ugm.v32f64.v32i1.v32i64(<32 x i1>, i8, i8, i8, i8, i8, i64, <32 x i64>, i16, i32, <32 x double>)
 
@@ -46,6 +47,26 @@ entry:
   ret <32 x i32> %res
 }
 
+; CHECK-LABEL: @test_fold_ascastp42p3_vector
+define <32 x double> @test_fold_ascastp42p3_vector(<32 x i1> %mask, <32 x double addrspace(4)*> %a) {
+entry:
+  %ascast = addrspacecast <32 x double addrspace(4)*> %a to <32 x double addrspace(3)*>
+  ; CHECK: %pti = ptrtoint <32 x double addrspace(3)*> %ascast to <32 x i32>
+  %pti = ptrtoint <32 x double addrspace(3)*> %ascast to <32 x i32>
+  %res = call <32 x double> @llvm.vc.internal.lsc.load.slm.v32f64.v32i1.v32i32(<32 x i1> %mask, i8 2, i8 4, i8 1, i8 0, i8 0, i32 0, <32 x i32> %pti, i16 1, i32 0, <32 x double> undef)
+  ret <32 x double> %res
+}
+
+; CHECK-LABEL: @test_fold_ascastp42p1_vector
+define <32 x double> @test_fold_ascastp42p1_vector(<32 x i1> %mask, <32 x double addrspace(4)*> %a) {
+entry:
+  %ascast = addrspacecast <32 x double addrspace(4)*> %a to <32 x double addrspace(1)*>
+  ; CHECK: %pti = ptrtoint <32 x double addrspace(4)*> %a to <32 x i64>
+  %pti = ptrtoint <32 x double addrspace(1)*> %ascast to <32 x i64>
+  %res = call <32 x double> @llvm.vc.internal.lsc.load.ugm.v32f64.v32i1.v32i64(<32 x i1> %mask, i8 3, i8 4, i8 1, i8 0, i8 0, i64 0, <32 x i64> %pti, i16 1, i32 0, <32 x double> undef)
+  ret <32 x double> %res
+}
+
 ; CHECK-LABEL: @test_fold_cast_vector
 define <32 x double> @test_fold_cast_vector(<32 x i1> %mask, <32 x i8 addrspace(1)*> %a) {
 entry: