llvm/test/CodeGen/SPIRV/expect.ll

; RUN: llc -verify-machineinstrs -mtriple=spirv32-unknown-unknown --spirv-ext=+SPV_KHR_expect_assume < %s | FileCheck --check-prefixes=CHECK,EXT %s
; RUN: llc -verify-machineinstrs -mtriple=spirv64-unknown-unknown --spirv-ext=+SPV_KHR_expect_assume < %s | FileCheck --check-prefixes=CHECK,EXT %s
; RUN: llc -verify-machineinstrs -mtriple=spirv32-unknown-unknown < %s | FileCheck --check-prefixes=CHECK,NOEXT %s
; RUN: llc -verify-machineinstrs -mtriple=spirv64-unknown-unknown < %s | FileCheck --check-prefixes=CHECK,NOEXT %s
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown --spirv-ext=+SPV_KHR_expect_assume < %s -o - -filetype=obj | spirv-val %}

; EXT:      OpCapability ExpectAssumeKHR
; EXT-NEXT: OpExtension "SPV_KHR_expect_assume"
; NOEXT-NOT:  OpCapability ExpectAssumeKHR
; NOEXT-NOT:  OpExtension "SPV_KHR_expect_assume"

declare i32 @llvm.expect.i32(i32, i32)
declare i32 @getOne()

; CHECK-DAG: %2 = OpTypeInt 32 0
; CHECK-DAG: %6 = OpFunctionParameter %2
; CHECK-DAG: %9 = OpIMul %2 %6 %8
; EXT-DAG:   %10 = OpExpectKHR %2 %9 %6
; NOEXT-NOT: %10 = OpExpectKHR %2 %9 %6

define i32 @test(i32 %x) {
  %one = call i32 @getOne()
  %val = mul i32 %x, %one
  %v = call i32 @llvm.expect.i32(i32 %val, i32 %x)
  ret i32 %v
}

; CHECK:     OpFunction
; CHECK:     %[[#P:]] = OpFunctionParameter %2
; CHECK:     %[[#V:]] = OpIMul %2 %[[#P]] %[[#]]
; EXT:       %[[#]] = OpExpectKHR %2 %[[#V]] %[[#P]]
; NOEXT-NOT: OpExpectKHR
; CHECK:     OpReturnValue

define i32 @test_with_probability(i32 %x) {
  %one = call i32 @getOne()
  %val = mul i32 %x, %one
  %v = call i32 @llvm.expect.with.probability.i32(i32 %val, i32 %x, double 0.9)
  ret i32 %v
}

; CHECK:     OpFunction
; CHECK:     %[[#P:]] = OpFunctionParameter %2
; CHECK:     %[[#V:]] = OpIMul %2 %[[#P]] %[[#]]
; CHECK-NOT: OpExpectKHR
; CHECK:     OpReturnValue %[[#V]]

define i32 @test_with_low_probability(i32 %x) {
  %one = call i32 @getOne()
  %val = mul i32 %x, %one
  %v = call i32 @llvm.expect.with.probability.i32(i32 %val, i32 %x, double 0.1)
  ret i32 %v
}