Lines Matching refs:gpu

7 // CHECK-LABEL: gpu.module @test {
8 gpu.module @test {
9 // CHECK: gpu.func @test_create_nd_tdesc_vc_1(%[[arg0:.*]]: memref<24x32xf32>) {
10 gpu.func @test_create_nd_tdesc_vc_1(%src: memref<24x32xf32>) {
13   gpu.return
16 // CHECK: gpu.func @test_create_nd_tdesc_with_sg_map(%[[arg0:.*]]: memref<24x32xf32>) {
17 gpu.func @test_create_nd_tdesc_with_sg_map(%src: memref<24x32xf32>) {
21   gpu.return
24 // CHECK: gpu.func @test_create_nd_tdesc_vc_2(%[[arg0:.*]]: ui64, %[[arg1:.*]]: index, %[[arg2:.*]]: index, %[[arg3:.*]]: index, %[[arg4:.*]]: index) {
25 gpu.func @test_create_nd_tdesc_vc_2(%src: ui64, %w : index, %h : index, %x : index, %y : index) {
30   gpu.return
33 // CHECK: gpu.func @test_create_nd_tdesc_vc_3(%[[arg0:.*]]: memref<24x32xf32>) {
34 gpu.func @test_create_nd_tdesc_vc_3(%src: memref<24x32xf32>) {
37   gpu.return
40 // CHECK: gpu.func @test_create_nd_tdesc_vc_4(%[[arg0:.*]]: memref<2x24x32xf32>) {
41 gpu.func @test_create_nd_tdesc_vc_4(%src: memref<2x24x32xf32>) {
44   gpu.return
47 // CHECK: gpu.func @test_create_nd_tdesc_vc_5(%[[arg0:.*]]: memref<2x24x32xf32, 3>) {
48 gpu.func @test_create_nd_tdesc_vc_5(%src: memref<2x24x32xf32, 3>) {
51   gpu.return
54 // CHECK: gpu.func @test_create_nd_tdesc_vc_6(%[[arg0:.*]]: memref<24x32xf32>) {
55 gpu.func @test_create_nd_tdesc_vc_6(%src: memref<24x32xf32>) {
58   gpu.return
61 // CHECK: gpu.func @test_prefetch_nd_vc(%[[arg0:.*]]: memref<24x32xf16>) {
62 gpu.func @test_prefetch_nd_vc(%src: memref<24x32xf16>) {
67   gpu.return
71 gpu.func @test_load_nd_vc(%src: memref<8x16xf16>) {
77   gpu.return
81 gpu.func @test_load_nd_vc_2(%src: memref<8x16xf16>) {
86   gpu.return
91 gpu.func @test_load_nd_vc_3(%src: memref<24x32xf32>) {
97   gpu.return
101 gpu.func @test_store_nd_vc(%dst: memref<24x32xf16>) {
108   gpu.return
112 gpu.func @test_store_nd_vc_2(%dst: memref<24x32xf16>) {
119   gpu.return
124 gpu.func @test_store_nd_vc_3(%src: memref<24x32xf16>) {
132   gpu.return
135 // CHECK: gpu.func @test_create_update_nd_tdesc_vc(%[[arg0:.*]]: memref<24x32xf32>) {
136 gpu.func @test_create_update_nd_tdesc_vc(%src: memref<24x32xf32>) {
141   gpu.return
144 // CHECK: gpu.func @test_create_tdesc_vc(%[[arg0:.*]]: ui64) {
145 gpu.func @test_create_tdesc_vc(%src: ui64) {
150   gpu.return
153 // CHECK: gpu.func @test_create_tdesc_vc_1(%[[arg0:.*]]: memref<?xf32, 3>) {
154 gpu.func @test_create_tdesc_vc_1(%src: memref<?xf32, 3>) {
159   gpu.return
162 // CHECK: gpu.func @test_create_tdesc_vc_with_sg_map(%[[arg0:.*]]: ui64) {
163 gpu.func @test_create_tdesc_vc_with_sg_map(%src: ui64) {
168   gpu.return
171 // CHECK: gpu.func @test_prefetch_vc(%[[arg0:.*]]: ui64) {
172 gpu.func @test_prefetch_vc(%src: ui64) {
179   gpu.return
182 // CHECK: gpu.func @test_load_gather_vc(%[[arg0:.*]]: ui64) {
183 gpu.func @test_load_gather_vc(%src: ui64) {
194   gpu.return
197 // CHECK: gpu.func @test_store_scatter_vc(%[[arg0:.*]]: ui64) {
198 gpu.func @test_store_scatter_vc(%src: ui64) {
211   gpu.return
214 // CHECK: gpu.func @test_create_update_tdesc_vc(%[[arg0:.*]]: ui64) {
215 gpu.func @test_create_update_tdesc_vc(%src: ui64) {
224   gpu.return
227 // CHECK: gpu.func @test_dpas_vc(%[[arg0:.*]]: vector<8x16xf16>, %[[arg1:.*]]: vector<16x16xf16>)
228 gpu.func @test_dpas_vc(%a : vector<8x16xf16>, %b: vector<16x16xf16>) {
231   gpu.return
235 // CHECK: gpu.func @test_dpas_vc_with_packed_b(%[[arg0:.*]]: vector<8x16xf16>, %[[arg1:.*]]: vector<8x16x2xf16>)
236 gpu.func @test_dpas_vc_with_packed_b(%a : vector<8x16xf16>, %b: vector<8x16x2xf16>) {
239   gpu.return
242 // CHECK: gpu.func @test_atomic_rmw(%[[arg0:.*]]: ui64, %[[arg1:.*]]: vector<16xf32>, %[[arg2:.*]]: vector<16xi1>)
243 gpu.func @test_atomic_rmw(%src: ui64, %value : vector<16xf32>, %mask : vector<16xi1>) {
250   gpu.return
253 // CHECK: gpu.func @alloc_nbarrier({{.*}}) {
254 gpu.func @alloc_nbarrier() {
257   gpu.return
260 // CHECK: gpu.func @init_nbarrier({{.*}}) {
261 gpu.func @init_nbarrier() {
268   gpu.return
271 // CHECK: gpu.func @nbarrier_arrive(%[[arg0:.*]]: !xegpu.nbarrier) {
272 gpu.func @nbarrier_arrive(%nbarrier : !xegpu.nbarrier) {
275   gpu.return
278 // CHECK: gpu.func @nbarrier_wait(%[[arg0:.*]]: !xegpu.nbarrier) {
279 gpu.func @nbarrier_wait(%nbarrier : !xegpu.nbarrier) {
282   gpu.return
285 // CHECK-LABEL: gpu.func @fence({{.*}}) {
286 gpu.func @fence() {
289   gpu.return