Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

backend: (csl) printing for functions, tasks and layout related things #2647

Merged
merged 20 commits into from
May 31, 2024
Merged

backend: (csl) printing for functions, tasks and layout related things #2647

Show file tree
Hide file tree
Changes from 16 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
4a0fbe2
backend: (csl) printing for functions and tasks
dk949 May 27, 2024
782d405
backend: (csl) printer for `csl.export`
dk949 May 27, 2024
4a7021b
backend: (csl) add printer for `csl.layout`
dk949 May 27, 2024
c95afde
backend: (csl) divider between layout and program
dk949 May 27, 2024
d127bec
fix: reintroduce removed newline before function def
dk949 May 27, 2024
4dca9c8
tests: (csl) ported over test from the old branch
dk949 May 27, 2024
b5c8fc2
tests: (csl) added tests for unbound tasks
dk949 May 27, 2024
1ee9891
fix: ssa name indices
dk949 May 27, 2024
2ee211b
added `_print_` prefix to printing functions
dk949 May 29, 2024
b94a8f9
`_get_layout_program` should take the module itself
dk949 May 29, 2024
91ea7de
better typing and documentation for `_print_task_or_fn`
dk949 May 29, 2024
0e4861b
documented `_get_layout_program`
dk949 May 29, 2024
ac4c44a
documented `_in_block_`
dk949 May 29, 2024
2dfbde7
documentation and better typing for `_wrapp_task_id`
dk949 May 29, 2024
cbaf24b
documentation for `_print_bind_task`
dk949 May 29, 2024
2f72b0a
documentation and better typing for `_memref_global_init`
dk949 May 29, 2024
da08a7f
simplified `Literal` type
dk949 May 29, 2024
fd3bdcc
fix: type in function name
dk949 May 29, 2024
eb27a2f
refactored descend as a context manager
dk949 May 29, 2024
0174108
removed unused `_in_block` method
dk949 May 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
243 changes: 235 additions & 8 deletions tests/filecheck/backend/csl/print_csl.mlir
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,106 @@
// RUN: xdsl-opt -t csl %s | filecheck %s

"csl.module"() <{kind=#csl<module_kind program>}> ({
// csl.func, csl.return
csl.func @no_args_no_return() {
csl.return
}

csl.func @no_args_return() -> f32 {
%c = arith.constant 5.0 : f32
csl.return %c : f32
}

csl.func @args_no_return(%a: i32, %b: i32) {
csl.return
}

// csl.const_struct
%empty_struct = "csl.const_struct"() : () -> !csl.comptime_struct
%attribute_struct = "csl.const_struct"() <{items = {hello = 123 : f32}}> : () -> !csl.comptime_struct
%const27 = arith.constant 27 : i16
%ssa_struct = "csl.const_struct"(%const27) <{ssa_fields = ["val"]}> : (i16) -> !csl.comptime_struct

%no_param_import = "csl.import_module"() <{module = "<mod>"}> : () -> !csl.imported_module
%param_import = "csl.import_module"(%ssa_struct) <{module = "<mod>"}> : (!csl.comptime_struct) -> !csl.imported_module

"csl.member_call"(%param_import) <{field = "foo"}> : (!csl.imported_module) -> ()
"csl.member_call"(%param_import, %const27) <{field = "bar"}> : (!csl.imported_module, i16) -> ()
%val2 = "csl.member_call"(%param_import) <{field = "baz"}> : (!csl.imported_module) -> (f32)

%val3 = "csl.member_access"(%param_import) <{field = "f"}> : (!csl.imported_module) -> (i32)


csl.func @main() {
"csl.call"() <{callee = @no_args_no_return}> : () -> ()
"csl.call"(%val3, %val3) <{callee = @args_no_return}> : (i32, i32) -> ()
%ret = "csl.call"() <{callee = @no_args_return}> : () -> (f32)

csl.return
}


csl.task @data_task(%arg: f32) attributes {kind = #csl<task_kind data>, id = 0 : i5} {
csl.return
}

csl.task @local_task() attributes {kind = #csl<task_kind local>, id = 1 : i5} {
csl.return
}

csl.task @control_task() attributes {kind = #csl<task_kind control>, id = 42 : i6} {
csl.return
}

csl.task @data_task_no_bind(%arg: f32) attributes {kind = #csl<task_kind data>} {
csl.return
}

csl.task @local_task_no_bind() attributes {kind = #csl<task_kind local>} {
csl.return
}

csl.task @control_task_no_bind() attributes {kind = #csl<task_kind control>} {
csl.return
}


"memref.global"() {"sym_name" = "uninit_array", "type" = memref<10xf32>, "sym_visibility" = "public", "initial_value"} : () -> ()
"memref.global"() {"sym_name" = "global_array", "type" = memref<10xf32>, "sym_visibility" = "public", "initial_value" = dense<4.2> : tensor<1xf32>} : () -> ()
"memref.global"() {"sym_name" = "const_array", "type" = memref<10xi32>, "sym_visibility" = "public", "constant", "initial_value" = dense<10> : tensor<1xi32>} : () -> ()


%uninit_array = memref.get_global @uninit_array : memref<10xf32>
%global_array = memref.get_global @global_array : memref<10xf32>
%const_array = memref.get_global @const_array : memref<10xi32>

%uninit_ptr = "csl.addressof"(%uninit_array) : (memref<10xf32>) -> !csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>
%global_ptr = "csl.addressof"(%global_array) : (memref<10xf32>) -> !csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>
%const_ptr = "csl.addressof"(%const_array) : (memref<10xi32>) -> !csl.ptr<i32, #csl<ptr_kind many>, #csl<ptr_const const>>

%ptr_to_arr = "csl.addressof"(%uninit_array) : (memref<10xf32>) -> !csl.ptr<memref<10xf32>, #csl<ptr_kind single>, #csl<ptr_const var>>
%ptr_to_val = "csl.addressof"(%const27) : (i16) -> !csl.ptr<i16, #csl<ptr_kind single>, #csl<ptr_const const>>


"csl.export"(%global_ptr) <{
type = !csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>,
var_name = "ptr_name"
}> : (!csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>) -> ()

"csl.export"(%const_ptr) <{
type = !csl.ptr<i32, #csl<ptr_kind many>, #csl<ptr_const const>>,
var_name = "another_ptr"
}> : (!csl.ptr<i32, #csl<ptr_kind many>, #csl<ptr_const const>>) -> ()

"csl.export"() <{type = () -> (), var_name = @no_args_no_return}> : () -> ()

"csl.export"() <{type = (i32, i32) -> (), var_name = @args_no_return}> : () -> ()

%col = "csl.get_color"() <{id = 15 : i5}> : () -> !csl.color

"csl.rpc"(%col) : (!csl.color) -> ()



"memref.global"() {"sym_name" = "A", "type" = memref<24xf32>, "sym_visibility" = "public", "initial_value" = dense<0> : tensor<1xindex>} : () -> ()
"memref.global"() {"sym_name" = "x", "type" = memref<6xf32>, "sym_visibility" = "public", "initial_value" = dense<0> : tensor<1xindex>} : () -> ()
Expand All @@ -9,6 +109,7 @@

%thing = "csl.import_module"() <{module = "<thing>"}> : () -> !csl.imported_module


csl.func @initialize() {
%lb = arith.constant 0 : i16
%ub = arith.constant 24 : i16
Expand Down Expand Up @@ -97,21 +198,128 @@ csl.func @gemv() {
}) {sym_name = "program"} : () -> ()


"csl.module"() <{kind=#csl<module_kind layout>}> ({
%p1 = "csl.param"() <{param_name = "param_1"}> : () -> i32
%p2 = "csl.param"() <{param_name = "param_2", init_value = 1.3 : f16}> : () -> f16

csl.layout {
%x_dim = arith.constant 4 : i32
%y_dim = arith.constant 6 : i32
"csl.set_rectangle"(%x_dim, %y_dim) : (i32, i32) -> ()

%x_coord0 = arith.constant 0 : i32
%y_coord = arith.constant 0 : i32
"csl.set_tile_code"(%x_coord0, %y_coord) <{file = "file.csl"}> : (i32, i32) -> ()

%params = "csl.const_struct"(){items = {hello = 123 : i32}} : () -> !csl.comptime_struct
%x_coord1 = arith.constant 1 : i32
"csl.set_tile_code"(%x_coord1, %y_coord, %params) <{file = "file.csl"}> : (i32, i32, !csl.comptime_struct) -> ()

}
}) {sym_name = "layout"} : () -> ()

// CHECK-NEXT:
// CHECK-NEXT: fn no_args_no_return() void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: fn no_args_return() f32 {
// CHECK-NEXT: const c : f32 = 5.0;
// CHECK-NEXT: return c;
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: fn args_no_return(a : i32, b : i32) void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT: //unknown op ConstStructOp(%empty_struct = "csl.const_struct"() : () -> !csl.comptime_struct)
// CHECK-NEXT: //unknown op ConstStructOp(%attribute_struct = "csl.const_struct"() <{"items" = {"hello" = 1.230000e+02 : f32}}> : () -> !csl.comptime_struct)
// CHECK-NEXT: const const27 : i16 = 27;
// CHECK-NEXT: //unknown op ConstStructOp(%ssa_struct = "csl.const_struct"(%const27) <{"ssa_fields" = ["val"]}> : (i16) -> !csl.comptime_struct)
// CHECK-NEXT: const no_param_import : imported_module = @import_module("<mod>");
// CHECK-NEXT: const param_import : imported_module = @import_module("<mod>", ssa_struct);
// CHECK-NEXT: param_import.foo();
// CHECK-NEXT: param_import.bar(const27);
// CHECK-NEXT: const val2 : f32 = param_import.baz();
// CHECK-NEXT: const val3 : i32 = param_import.f;
// CHECK-NEXT:
// CHECK-NEXT: fn main() void {
// CHECK-NEXT: no_args_no_return();
// CHECK-NEXT: args_no_return(val3, val3);
// CHECK-NEXT: const ret : f32 = no_args_return();
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: task data_task(arg : f32) void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT: comptime {
// CHECK-NEXT: @bind_data_task(data_task, @get_data_task_id(@get_color(0)));
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: task local_task() void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT: comptime {
// CHECK-NEXT: @bind_local_task(local_task, @get_local_task_id(1));
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: task control_task() void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT: comptime {
// CHECK-NEXT: @bind_control_task(control_task, @get_control_task_id(42));
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: task data_task_no_bind(arg0 : f32) void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: task local_task_no_bind() void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: task control_task_no_bind() void {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT: var uninit_array : [10]f32;
// CHECK-NEXT: var global_array : [10]f32 = @constants([10]f32, 4.2);
// CHECK-NEXT: const const_array : [10]i32 = @constants([10]i32, 10);

// CHECK-NEXT: //unknown op AddressOfOp(%uninit_ptr = "csl.addressof"(%uninit_array) : (memref<10xf32>) -> !csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>)
// CHECK-NEXT: //unknown op AddressOfOp(%global_ptr = "csl.addressof"(%global_array) : (memref<10xf32>) -> !csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>)
// CHECK-NEXT: //unknown op AddressOfOp(%const_ptr = "csl.addressof"(%const_array) : (memref<10xi32>) -> !csl.ptr<i32, #csl<ptr_kind many>, #csl<ptr_const const>>)
// CHECK-NEXT: //unknown op AddressOfOp(%ptr_to_arr = "csl.addressof"(%uninit_array) : (memref<10xf32>) -> !csl.ptr<memref<10xf32>, #csl<ptr_kind single>, #csl<ptr_const var>>)
// CHECK-NEXT: //unknown op AddressOfOp(%ptr_to_val = "csl.addressof"(%const27) : (i16) -> !csl.ptr<i16, #csl<ptr_kind single>, #csl<ptr_const const>>)
// CHECK-NEXT: comptime {
// CHECK-NEXT: @export_symbol(global_ptr, "ptr_name");
// CHECK-NEXT: }
// CHECK-NEXT: comptime {
// CHECK-NEXT: @export_symbol(const_ptr, "another_ptr");
// CHECK-NEXT: }
// CHECK-NEXT: comptime {
// CHECK-NEXT: @export_symbol(no_args_no_return, "no_args_no_return");
// CHECK-NEXT: }
// CHECK-NEXT: comptime {
// CHECK-NEXT: @export_symbol(args_no_return, "args_no_return");
// CHECK-NEXT: }
// CHECK-NEXT: //unknown op GetColorOp(%col = "csl.get_color"() <{"id" = 15 : i5}> : () -> !csl.color)
// CHECK-NEXT: //unknown op RpcOp("csl.rpc"(%col) : (!csl.color) -> ())

// CHECK-NEXT: var A : [24]f32 = @constants([24]f32, 0);
// CHECK-NEXT: var x : [6]f32 = @constants([6]f32, 0);
// CHECK-NEXT: var b : [4]f32 = @constants([4]f32, 0);
// CHECK-NEXT: var y : [4]f32 = @constants([4]f32, 0);
// CHECK-NEXT: const thing : imported_module = @import_module("<thing>");
// CHECK-NEXT:
// CHECK-NEXT: fn initialize() {
// CHECK-NEXT: fn initialize() void {
// CHECK-NEXT: const lb : i16 = 0;
// CHECK-NEXT: const ub : i16 = 24;
// CHECK-NEXT: const step : i16 = 1;
// CHECK-NEXT: thing.some_func(lb, ub);
// CHECK-NEXT: const res : i32 = thing.some_func(lb, ub);
// CHECK-NEXT: const v0 : comptime_struct = thing.some_field;
// CHECK-NEXT: const v1 : f32 = 3.14;
// CHECK-NEXT: const v02 : f16 = 2.718;
// CHECK-NEXT: const v1 : comptime_struct = thing.some_field;
// CHECK-NEXT: const v2 : f32 = 3.14;
// CHECK-NEXT: const v0 : f16 = 2.718;
// CHECK-NEXT: const u32cst : u32 = 44;
// CHECK-NEXT:
// CHECK-NEXT: for(@range(i16, lb, ub, step)) |idx| {
Expand All @@ -138,15 +346,15 @@ csl.func @gemv() {
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT:
// CHECK-NEXT: fn gemv() {
// CHECK-NEXT: fn gemv() void {
// CHECK-NEXT: const lb : i32 = 0;
// CHECK-NEXT: const step : i32 = 1;
// CHECK-NEXT: const ub : i32 = 6;
// CHECK-NEXT: const ub0 : i32 = 4;
// CHECK-NEXT: const ub1 : i32 = 4;
// CHECK-NEXT:
// CHECK-NEXT: for(@range(i32, lb, ub0, step)) |i| {
// CHECK-NEXT: for(@range(i32, lb, ub1, step)) |i| {
// CHECK-NEXT: const tmp : f32 = 0.0;
// CHECK-NEXT: var tmp1 : f32 = tmp;
// CHECK-NEXT: var tmp2 : f32 = tmp;
// CHECK-NEXT:
// CHECK-NEXT: for(@range(i32, lb, ub, step)) |j| {
// CHECK-NEXT: //unknown op Muli(%ix6 = arith.muli %i, %ub : index)
Expand All @@ -162,4 +370,23 @@ csl.func @gemv() {
// CHECK-NEXT: }
// CHECK-NEXT: return;
// CHECK-NEXT: }
// CHECK-NEXT: // >>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<< //
// CHECK-NEXT: //unknown op ParamOp(%p1 = "csl.param"() <{"param_name" = "param_1"}> : () -> i32)
// CHECK-NEXT: //unknown op ParamOp(%p2 = "csl.param"() <{"param_name" = "param_2", "init_value" = 1.300000e+00 : f16}> : () -> f16)
// CHECK-NEXT: layout {
// CHECK-NEXT: const x_dim : i32 = 4;
// CHECK-NEXT: const y_dim : i32 = 6;
// CHECK-NEXT: //unknown op SetRectangleOp("csl.set_rectangle"(%x_dim, %y_dim) : (i32, i32) -> ())
// CHECK-NEXT: const x_coord0 : i32 = 0;
// CHECK-NEXT: const y_coord : i32 = 0;
// CHECK-NEXT: //unknown op SetTileCodeOp("csl.set_tile_code"(%x_coord0, %y_coord) <{"file" = "file.csl"}> : (i32, i32) -> ())
// CHECK-NEXT: //unknown op ConstStructOp(%params = "csl.const_struct"() <{"items" = {"hello" = 123 : i32}}> : () -> !csl.comptime_struct)
// CHECK-NEXT: const x_coord1 : i32 = 1;
// CHECK-NEXT: //unknown op SetTileCodeOp("csl.set_tile_code"(%x_coord1, %y_coord, %params) <{"file" = "file.csl"}> : (i32, i32, !csl.comptime_struct) -> ())
// CHECK-NEXT: @export_name("ptr_name", <!unknown value !csl.ptr<f32, #csl<ptr_kind many>, #csl<ptr_const var>>>, true);
// CHECK-NEXT: @export_name("another_ptr", <!unknown value !csl.ptr<i32, #csl<ptr_kind many>, #csl<ptr_const const>>>, false);
// CHECK-NEXT: @export_name("no_args_no_return", <!unknown value () -> ()>, );
// CHECK-NEXT: @export_name("args_no_return", <!unknown value (i32, i32) -> ()>, );
// CHECK-NEXT: }

// CHECK-EMPTY:
Loading
Loading