derek/TheRock
1
0 Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
1a182e2dac58879f215c0fad0f5f78c76dcafe56
TheRock/compiler/amd-llvm/mlir/test/Conversion/SPIRVToLLVM/gl-ops-to-llvm.mlir
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
Details
CodeQL / Analyze (python) (push) Canceled after 0s
Details
pre-commit / pre-commit (push) Canceled after 0s
Details
Multi-Arch CI / setup (push) Canceled after 0s
Details
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Details
Setup (Multi-Arch) / setup (push) Canceled after 0s
Details
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Details
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
Details
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Details
Multi-Arch CI / CI Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Details
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Details
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

336 lines
12 KiB
MLIR
Raw Blame History

// RUN: mlir-opt -convert-spirv-to-llvm %s | FileCheck %s
//===----------------------------------------------------------------------===//
// spirv.GL.Ceil
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @ceil
spirv.func @ceil(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.ceil(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Ceil %arg0 : f32
// CHECK: llvm.intr.ceil(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Ceil %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Cos
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @cos
spirv.func @cos(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.cos(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Cos %arg0 : f32
// CHECK: llvm.intr.cos(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Cos %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Exp
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @exp
spirv.func @exp(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.exp(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Exp %arg0 : f32
// CHECK: llvm.intr.exp(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Exp %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.FAbs
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @fabs
spirv.func @fabs(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.fabs(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.FAbs %arg0 : f32
// CHECK: llvm.intr.fabs(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.FAbs %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Floor
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @floor
spirv.func @floor(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.floor(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Floor %arg0 : f32
// CHECK: llvm.intr.floor(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Floor %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.FMax
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @fmax
spirv.func @fmax(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.maxnum(%{{.*}}, %{{.*}}) : (f32, f32) -> f32
%0 = spirv.GL.FMax %arg0, %arg0 : f32
// CHECK: llvm.intr.maxnum(%{{.*}}, %{{.*}}) : (vector<3xf16>, vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.FMax %arg1, %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.FMin
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @fmin
spirv.func @fmin(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.minnum(%{{.*}}, %{{.*}}) : (f32, f32) -> f32
%0 = spirv.GL.FMin %arg0, %arg0 : f32
// CHECK: llvm.intr.minnum(%{{.*}}, %{{.*}}) : (vector<3xf16>, vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.FMin %arg1, %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Log
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @log
spirv.func @log(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.log(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Log %arg0 : f32
// CHECK: llvm.intr.log(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Log %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Sin
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @sin
spirv.func @sin(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.sin(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Sin %arg0 : f32
// CHECK: llvm.intr.sin(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Sin %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Pow
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @pow
spirv.func @pow(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.pow(%{{.*}}, %{{.*}}) : (f32, f32) -> f32
%0 = spirv.GL.Pow %arg0, %arg0 : f32
// CHECK: llvm.intr.pow(%{{.*}}, %{{.*}}) : (vector<3xf16>, vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Pow %arg1, %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Fma
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @fma
spirv.func @fma(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.fma(%{{.*}}, %{{.*}}, %{{.*}}) : (f32, f32, f32) -> f32
%0 = spirv.GL.Fma %arg0, %arg0, %arg0 : f32
// CHECK: llvm.intr.fma(%{{.*}}, %{{.*}}, %{{.*}}) : (vector<3xf16>, vector<3xf16>, vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Fma %arg1, %arg1, %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.SAbs
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @sabs
spirv.func @sabs(%arg0: i16, %arg1: vector<3xi32>) "None" {
// CHECK: "llvm.intr.abs"(%{{.*}}) <{is_int_min_poison = false}> : (i16) -> i16
%0 = spirv.GL.SAbs %arg0 : i16
// CHECK: "llvm.intr.abs"(%{{.*}}) <{is_int_min_poison = false}> : (vector<3xi32>) -> vector<3xi32>
%1 = spirv.GL.SAbs %arg1 : vector<3xi32>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.SMax
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @smax
spirv.func @smax(%arg0: i16, %arg1: vector<3xi32>) "None" {
// CHECK: llvm.intr.smax(%{{.*}}, %{{.*}}) : (i16, i16) -> i16
%0 = spirv.GL.SMax %arg0, %arg0 : i16
// CHECK: llvm.intr.smax(%{{.*}}, %{{.*}}) : (vector<3xi32>, vector<3xi32>) -> vector<3xi32>
%1 = spirv.GL.SMax %arg1, %arg1 : vector<3xi32>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.SMin
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @smin
spirv.func @smin(%arg0: i16, %arg1: vector<3xi32>) "None" {
// CHECK: llvm.intr.smin(%{{.*}}, %{{.*}}) : (i16, i16) -> i16
%0 = spirv.GL.SMin %arg0, %arg0 : i16
// CHECK: llvm.intr.smin(%{{.*}}, %{{.*}}) : (vector<3xi32>, vector<3xi32>) -> vector<3xi32>
%1 = spirv.GL.SMin %arg1, %arg1 : vector<3xi32>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.UMax
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @umax
spirv.func @umax(%arg0: i16, %arg1: vector<3xi32>) "None" {
// CHECK: llvm.intr.umax(%{{.*}}, %{{.*}}) : (i16, i16) -> i16
%0 = spirv.GL.UMax %arg0, %arg0 : i16
// CHECK: llvm.intr.umax(%{{.*}}, %{{.*}}) : (vector<3xi32>, vector<3xi32>) -> vector<3xi32>
%1 = spirv.GL.UMax %arg1, %arg1 : vector<3xi32>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.UMin
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @umin
spirv.func @umin(%arg0: i16, %arg1: vector<3xi32>) "None" {
// CHECK: llvm.intr.umin(%{{.*}}, %{{.*}}) : (i16, i16) -> i16
%0 = spirv.GL.UMin %arg0, %arg0 : i16
// CHECK: llvm.intr.umin(%{{.*}}, %{{.*}}) : (vector<3xi32>, vector<3xi32>) -> vector<3xi32>
%1 = spirv.GL.UMin %arg1, %arg1 : vector<3xi32>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Sqrt
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @sqrt
spirv.func @sqrt(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.sqrt(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Sqrt %arg0 : f32
// CHECK: llvm.intr.sqrt(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Sqrt %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Tan
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @tan
spirv.func @tan(%arg0: f32) "None" {
// CHECK: llvm.intr.tan(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Tan %arg0 : f32
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Tanh
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @tanh
spirv.func @tanh(%arg0: f32) "None" {
// CHECK: llvm.intr.tanh(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Tanh %arg0 : f32
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Exp2
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @exp2
spirv.func @exp2(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.exp2(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Exp2 %arg0 : f32
// CHECK: llvm.intr.exp2(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Exp2 %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Log2
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @log2
spirv.func @log2(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.log2(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Log2 %arg0 : f32
// CHECK: llvm.intr.log2(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Log2 %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Round
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @round
spirv.func @round(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.round(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Round %arg0 : f32
// CHECK: llvm.intr.round(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Round %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.RoundEven
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @round_even
spirv.func @round_even(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.roundeven(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.RoundEven %arg0 : f32
// CHECK: llvm.intr.roundeven(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.RoundEven %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Sinh
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @sinh
spirv.func @sinh(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.sinh(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Sinh %arg0 : f32
// CHECK: llvm.intr.sinh(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Sinh %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.Cosh
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @cosh
spirv.func @cosh(%arg0: f32, %arg1: vector<3xf16>) "None" {
// CHECK: llvm.intr.cosh(%{{.*}}) : (f32) -> f32
%0 = spirv.GL.Cosh %arg0 : f32
// CHECK: llvm.intr.cosh(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>
%1 = spirv.GL.Cosh %arg1 : vector<3xf16>
spirv.Return
}
//===----------------------------------------------------------------------===//
// spirv.GL.InverseSqrt
//===----------------------------------------------------------------------===//
// CHECK-LABEL: @inverse_sqrt
spirv.func @inverse_sqrt(%arg0: f32) "None" {
// CHECK: %[[ONE:.*]] = llvm.mlir.constant(1.000000e+00 : f32) : f32
// CHECK: %[[SQRT:.*]] = llvm.intr.sqrt(%{{.*}}) : (f32) -> f32
// CHECK: llvm.fdiv %[[ONE]], %[[SQRT]] : f32
%0 = spirv.GL.InverseSqrt %arg0 : f32
spirv.Return
}
Reference in new issue View Git Blame Copy Permalink