| // RUN: mlir-opt -convert-xegpu-to-xevm %s | FileCheck %s |
| |
| gpu.module @test_kernel [#xevm.target<chip = "pvc">] { |
| // CHECK-LABEL: func.func @dpas( |
| // CHECK-SAME: %[[ARG0:.*]]: vector<8xf16>, %[[ARG1:.*]]: vector<16xf16>, %[[ARG2:.*]]: vector<8xf32> |
| func.func @dpas(%a_loaded: vector<8xf16>, %b_loaded: vector<16xf16>, %c_loaded: vector<8xf32>) -> vector<8xf32> { |
| // Loads are checked in a separate test. |
| // CHECK: %[[D:.*]] = xevm.mma %[[ARG0]], %[[ARG1]], %[[ARG2]] {shape = <m = 8, n = 16, k = 16>, types = <d = f32, a = f16, b = f16, c = f32>} |
| // CHECK-SAME: : (vector<8xf16>, vector<16xf16>, vector<8xf32>) -> vector<8xf32> |
| %d = xegpu.dpas %a_loaded, %b_loaded, %c_loaded |
| : vector<8xf16>, vector<16xf16>, vector<8xf32> -> vector<8xf32> |
| return %d : vector<8xf32> |
| } |
| } |
