-
Notifications
You must be signed in to change notification settings - Fork 47
Expand file tree
/
Copy pathalloc_tile_addr_dynamic_no_rebind.pto
More file actions
23 lines (19 loc) · 1.07 KB
/
alloc_tile_addr_dynamic_no_rebind.pto
File metadata and controls
23 lines (19 loc) · 1.07 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
// RUN: ptoas --pto-level=level3 %s | FileCheck %s
module {
func.func @print_alloc_addr_dyn(%arg0: index, %arg1: index) attributes {pto.entry} {
%c0_i64 = arith.constant 0 : i64
%0 = pto.alloc_tile addr = %c0_i64 valid_row = %arg0 valid_col = %arg1
: !pto.tile_buf<loc=vec, dtype=f16, rows=64, cols=128, v_row=?, v_col=?,
blayout=row_major, slayout=none_box, fractal=512, pad=0>
pto.tprint ins(%0 : !pto.tile_buf<loc=vec, dtype=f16, rows=64, cols=128,
v_row=?, v_col=?, blayout=row_major,
slayout=none_box, fractal=512, pad=0>)
return
}
}
// CHECK-LABEL: __global__ AICORE void print_alloc_addr_dyn(
// CHECK: Tile<TileType::Vec, half, 64, 128, BLayout::RowMajor, -1, -1, SLayout::NoneBox, 512, PadValue::Null> [[TILE:v[0-9]+]] = Tile<TileType::Vec, half, 64, 128, BLayout::RowMajor, -1, -1, SLayout::NoneBox, 512, PadValue::Null>(
// CHECK: TASSIGN([[TILE]], [[ADDR:v[0-9]+]]);
// CHECK-NOT: .data()
// CHECK-NOT: reinterpret_cast<uint64_t>
// CHECK: TPRINT([[TILE]]);