Skip to content

Commit 85e598c

Browse files
authored
add amdgpu runtime example (#132)
1 parent 830c7e1 commit 85e598c

File tree

4 files changed

+1301
-11
lines changed

4 files changed

+1301
-11
lines changed

mlir/extras/runtime/passes.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def add_pass(self, pass_name, **kwargs):
135135
self._pipeline.append(pass_str)
136136
return self
137137

138-
def lower_to_llvm(self):
138+
def lower_to_llvm(self, use_bare_ptr_memref_call_conv=False):
139139
# https://github.com/makslevental/llvm-project/blob/f6643263631bcb0d191ef923963ac1a5ca9ac5fd/mlir/test/lib/Dialect/LLVM/TestLowerToLLVM.cpp#L44
140140
return (
141141
self.Func(
@@ -163,7 +163,9 @@ def lower_to_llvm(self):
163163
# Convert MemRef to LLVM (always needed).
164164
.finalize_memref_to_llvm()
165165
# Convert Func to LLVM (always needed).
166-
.convert_func_to_llvm()
166+
.convert_func_to_llvm(
167+
use_bare_ptr_memref_call_conv=use_bare_ptr_memref_call_conv
168+
)
167169
.convert_arith_to_llvm()
168170
.convert_cf_to_llvm()
169171
# Convert Index to LLVM (always needed).

0 commit comments

Comments
 (0)