Created
November 4, 2024 19:09
-
-
Save pashu123/0b5329e8a60311b634ce9d2230381b20 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#pipeline_layout = #hal.pipeline.layout<constants = 4, bindings = [ | |
#hal.pipeline.binding<storage_buffer> | |
]> | |
hal.executable.source public @executable { | |
hal.executable.export public @write_constants ordinal(0) layout(#pipeline_layout) attributes {workgroup_size = [1 : index, 1 : index, 1 : index]} { | |
^bb0(%arg0: !hal.device): | |
%c1 = arith.constant 1 : index | |
hal.return %c1, %c1, %c1 : index, index, index | |
} | |
builtin.module { | |
func.func @write_constants() { | |
%input_0 = hal.interface.constant.load layout(#pipeline_layout) ordinal(0) : i32 | |
%input_1 = hal.interface.constant.load layout(#pipeline_layout) ordinal(1) : i32 | |
%input_2 = hal.interface.constant.load layout(#pipeline_layout) ordinal(2) : i32 | |
%input_3 = hal.interface.constant.load layout(#pipeline_layout) ordinal(3) : i32 | |
%out = hal.interface.binding.subspan layout(#pipeline_layout) binding(0) : memref<4xi32> | |
%c0 = arith.constant 0 : index | |
%c1 = arith.constant 1 : index | |
%c2 = arith.constant 2 : index | |
%c3 = arith.constant 3 : index | |
memref.store %input_0, %out[%c0] : memref<4xi32> | |
memref.store %input_1, %out[%c1] : memref<4xi32> | |
memref.store %input_2, %out[%c2] : memref<4xi32> | |
memref.store %input_3, %out[%c3] : memref<4xi32> | |
return | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment