llvm-project/llvm/test/Transforms/HotColdSplit/assumption-cache-invalidation.ll
Kajetan Puchalski d3ed84ed67
[Utils][mlir] Fix interaction between CodeExtractor and OpenMPIRBuilder (#145051)
CodeExtractor can currently erroneously insert an alloca into a
different function than it inserts its users into, in cases where code
is being extracted out of a function that has already been outlined. Add
an assertion that the two blocks being inserted into are actually in the
same function.

Add a check to findAllocaInsertPoint in OpenMP to LLVMIR translation to
prevent the aforementioned scenario from happening.

OpenMPIRBuilder relies on a callback mechanism to fix-up a module later
on during the finaliser step. In some cases this results in the module
being invalid prior to the finalise step running. Remove calls to
verifyModule wrapped in LLVM_DEBUG from CodeExtractor, as the presence
of those results in the compiler crashing with -mllvm -debug due to
premature module verification where it would not crash without -debug.

Call ompBuilder->finalize() the end of mlir::translateModuleToLLVMIR, in
order to make sure the module has actually been finalized prior to
trying to verify it.

Resolves https://github.com/llvm/llvm-project/issues/138102.

---------

Signed-off-by: Kajetan Puchalski <kajetan.puchalski@arm.com>
2025-06-25 13:34:35 +01:00

54 lines
1.6 KiB
LLVM

; REQUIRES: asserts
; RUN: opt -S -passes='function(instsimplify),hotcoldsplit' -hotcoldsplit-threshold=-1 < %s 2>&1 | FileCheck %s
; RUN: opt -passes='function(instcombine),hotcoldsplit,function(instsimplify)' %s -o /dev/null
target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
target triple = "aarch64"
%a = type { i64, i64 }
%b = type { i64 }
; CHECK: @f
; CHECK-LABEL: codeRepl:
; CHECK-NOT: @llvm.assume
; CHECK: }
; CHECK: declare {{.*}}@llvm.assume
; CHECK: define {{.*}}@f.cold.1(i64 %load1)
; CHECK-LABEL: newFuncRoot:
; CHECK: %cmp1 = icmp eq i64 %load1, 0
; CHECK-NOT: call void @llvm.assume
; CHECK: define {{.*}}@f.cold.2()
; CHECK-LABEL: newFuncRoot:
; CHECK: }
define void @f() {
entry:
%i = getelementptr inbounds %a, ptr null, i64 0, i32 1
br label %label
label: ; preds = %entry
%load0 = load ptr, ptr %i, align 8
%i3 = getelementptr inbounds %b, ptr %load0, i64 undef, i32 0
%load1 = load i64, ptr %i3, align 8
%cmp0 = icmp ugt i64 %load1, 1
br i1 %cmp0, label %if.then, label %if.else
if.then: ; preds = %label
unreachable
if.else: ; preds = %label
call void @g(ptr undef)
%load2 = load i64, ptr undef, align 8
%i7 = and i64 %load2, -16
%i8 = inttoptr i64 %i7 to ptr
%cmp1 = icmp eq i64 %load1, 0
call void @llvm.assume(i1 %cmp1)
unreachable
}
declare void @g(ptr)
declare void @llvm.assume(i1 noundef) #0
attributes #0 = { nocallback nofree nosync nounwind willreturn memory(inaccessiblemem: readwrite) }