derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
136 lines
8.0 KiB
MLIR
136 lines
8.0 KiB
MLIR
|
|
// RUN: mlir-translate --no-implicit-module --test-spirv-roundtrip %s | FileCheck %s
|
|
// RUN: %if spirv-tools %{ mlir-translate --no-implicit-module --serialize-spirv %s | spirv-val %}
|
|
|
|
// Note: Since the output of this test (optionally) gets validated by spirv-val,
|
|
// we cannot use splits.
|
|
|
|
spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, ReplicatedCompositesEXT, TensorsARM, Linkage],
|
|
[SPV_EXT_replicated_composites, SPV_ARM_tensors]> {
|
|
// CHECK-LABEL: @splat_vector_i32
|
|
spirv.func @splat_vector_i32() -> (vector<3xi32>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [1 : i32] : vector<3xi32>
|
|
%1 = spirv.EXT.ConstantCompositeReplicate [1 : i32] : vector<3xi32>
|
|
spirv.ReturnValue %1 : vector<3xi32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_i32
|
|
spirv.func @splat_array_of_i32() -> (!spirv.array<3 x i32>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [1 : i32] : !spirv.array<3 x i32>
|
|
%1 = spirv.EXT.ConstantCompositeReplicate [1 : i32] : !spirv.array<3 x i32>
|
|
spirv.ReturnValue %1 : !spirv.array<3 x i32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_vectors_of_i32
|
|
spirv.func @splat_array_of_vectors_of_i32() -> (!spirv.array<3 x vector<2xi32>>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [dense<[1, 2]> : vector<2xi32>] : !spirv.array<3 x vector<2xi32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [dense<[1, 2]> : vector<2xi32>] : !spirv.array<3 x vector<2xi32>>
|
|
spirv.ReturnValue %0 : !spirv.array<3 x vector<2xi32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_splat_array_of_i32
|
|
spirv.func @splat_array_of_splat_array_of_i32() -> (!spirv.array<2 x !spirv.array<3 x i32>>) "None" {
|
|
// CHECK: %0 = spirv.EXT.ConstantCompositeReplicate [3 : i32] : !spirv.array<2 x !spirv.array<3 x i32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [3 : i32] : !spirv.array<2 x !spirv.array<3 x i32>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x !spirv.array<3 x i32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_non_splat_array_of_i32
|
|
spirv.func @splat_array_of_non_splat_array_of_i32() -> (!spirv.array<2 x !spirv.array<3 x i32>>) "None" {
|
|
// CHECK: %0 = spirv.EXT.ConstantCompositeReplicate {{\[}}[1 : i32, 2 : i32, 3 : i32]] : !spirv.array<2 x !spirv.array<3 x i32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [[1 : i32, 2 : i32, 3 : i32]] : !spirv.array<2 x !spirv.array<3 x i32>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x !spirv.array<3 x i32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_splat_vectors_of_i32
|
|
spirv.func @splat_array_of_splat_vectors_of_i32() -> (!spirv.array<2 x vector<2xi32>>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [2 : i32] : !spirv.array<2 x vector<2xi32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [2 : i32] : !spirv.array<2 x vector<2xi32>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x vector<2xi32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_arm_tensor_of_i32
|
|
spirv.func @splat_arm_tensor_of_i32() -> (!spirv.arm.tensor<2x3xi32>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [2 : i32] : !spirv.arm.tensor<2x3xi32>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [2 : i32] : !spirv.arm.tensor<2x3xi32>
|
|
spirv.ReturnValue %0 : !spirv.arm.tensor<2x3xi32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_non_splat_array_of_arrays_of_i32
|
|
spirv.func @splat_array_of_non_splat_array_of_arrays_of_i32() -> !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x i32>>> "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate {{\[}}{{\[}}[1 : i32, 2 : i32, 3 : i32], [4 : i32, 5 : i32, 6 : i32]]] : !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x i32>>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [[[1 : i32, 2 : i32, 3 : i32], [4 : i32, 5 : i32, 6 : i32]]] : !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x i32>>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x i32>>>
|
|
}
|
|
|
|
// CHECK-LABEL: @null_cc_arm_tensor_of_i32
|
|
spirv.func @null_cc_arm_tensor_of_i32() -> (!spirv.arm.tensor<2x3xi32>) "None" {
|
|
// CHECK: spirv.Constant dense<0> : !spirv.arm.tensor<2x3xi32>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [0 : i32] : !spirv.arm.tensor<2x3xi32>
|
|
spirv.ReturnValue %0 : !spirv.arm.tensor<2x3xi32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_vector_f32
|
|
spirv.func @splat_vector_f32() -> (vector<3xf32>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [1.000000e+00 : f32] : vector<3xf32>
|
|
%1 = spirv.EXT.ConstantCompositeReplicate [1.0 : f32] : vector<3xf32>
|
|
spirv.ReturnValue %1 : vector<3xf32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_f32
|
|
spirv.func @splat_array_of_f32() -> (!spirv.array<3 x f32>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [1.000000e+00 : f32] : !spirv.array<3 x f32>
|
|
%1 = spirv.EXT.ConstantCompositeReplicate [1.0 : f32] : !spirv.array<3 x f32>
|
|
spirv.ReturnValue %1 : !spirv.array<3 x f32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_splat_array_of_f32
|
|
spirv.func @splat_array_of_splat_array_of_f32() -> (!spirv.array<2 x !spirv.array<3 x f32>>) "None" {
|
|
// CHECK: %0 = spirv.EXT.ConstantCompositeReplicate [3.000000e+00 : f32] : !spirv.array<2 x !spirv.array<3 x f32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [3.0 : f32] : !spirv.array<2 x !spirv.array<3 x f32>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x !spirv.array<3 x f32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_non_splat_array_of_f32
|
|
spirv.func @splat_array_of_non_splat_array_of_f32() -> (!spirv.array<2 x !spirv.array<3 x f32>>) "None" {
|
|
// CHECK: %0 = spirv.EXT.ConstantCompositeReplicate {{\[}}[1.000000e+00 : f32, 2.000000e+00 : f32, 3.000000e+00 : f32]] : !spirv.array<2 x !spirv.array<3 x f32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [[1.0 : f32, 2.0 : f32, 3.0 : f32]] : !spirv.array<2 x !spirv.array<3 x f32>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x !spirv.array<3 x f32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_vectors_of_f32
|
|
spirv.func @splat_array_of_vectors_of_f32() -> (!spirv.array<3 x vector<2xf32>>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [dense<[1.000000e+00, 2.000000e+00]> : vector<2xf32>] : !spirv.array<3 x vector<2xf32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [dense<[1.0, 2.0]> : vector<2xf32>] : !spirv.array<3 x vector<2xf32>>
|
|
spirv.ReturnValue %0 : !spirv.array<3 x vector<2xf32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_splat_vectors_of_f32
|
|
spirv.func @splat_array_of_splat_vectors_of_f32() -> (!spirv.array<2 x vector<2xf32>>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [2.000000e+00 : f32] : !spirv.array<2 x vector<2xf32>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [2.0 : f32] : !spirv.array<2 x vector<2xf32>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x vector<2xf32>>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_arm_tensor_of_f32
|
|
spirv.func @splat_arm_tensor_of_f32() -> (!spirv.arm.tensor<2x3xf32>) "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate [2.000000e+00 : f32] : !spirv.arm.tensor<2x3xf32>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [2.0 : f32] : !spirv.arm.tensor<2x3xf32>
|
|
spirv.ReturnValue %0 : !spirv.arm.tensor<2x3xf32>
|
|
}
|
|
|
|
// CHECK-LABEL: @splat_array_of_non_splat_array_of_arrays_of_f32
|
|
spirv.func @splat_array_of_non_splat_array_of_arrays_of_f32() -> !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x f32>>> "None" {
|
|
// CHECK: spirv.EXT.ConstantCompositeReplicate {{\[}}{{\[}}[1.000000e+00 : f32, 2.000000e+00 : f32, 3.000000e+00 : f32], [4.000000e+00 : f32, 5.000000e+00 : f32, 6.000000e+00 : f32]]] : !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x f32>>>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [[[1.0 : f32, 2.0 : f32, 3.0 : f32], [4.0 : f32, 5.0 : f32, 6.0 : f32]]] : !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x f32>>>
|
|
spirv.ReturnValue %0 : !spirv.array<2 x !spirv.array<2 x !spirv.array<3 x f32>>>
|
|
}
|
|
|
|
// CHECK-LABEL: @null_cc_arm_tensor_of_f32
|
|
spirv.func @null_cc_arm_tensor_of_f32() -> (!spirv.arm.tensor<2x3xf32>) "None" {
|
|
// CHECK: spirv.Constant dense<0.000000e+00> : !spirv.arm.tensor<2x3xf32>
|
|
%0 = spirv.EXT.ConstantCompositeReplicate [0.0 : f32] : !spirv.arm.tensor<2x3xf32>
|
|
spirv.ReturnValue %0 : !spirv.arm.tensor<2x3xf32>
|
|
}
|
|
}
|