Skip to content

Commit 4fb023f

Browse files
adam-smnkkurapov-peter
authored andcommitted
Fix return type
1 parent 0a25818 commit 4fb023f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/examples/mlir/compile_and_run.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def create_schedule(ctx: ir.Context) -> ir.Module:
104104
return schedule
105105

106106

107-
def apply_schedule(kernel: ir.Module, schedule: ir.Module) -> ir.Module:
107+
def apply_schedule(kernel: ir.Module, schedule: ir.Module) -> None:
108108
"""
109109
Apply transformation schedule to a kernel module.
110110
The kernel is modified in-place.
@@ -135,6 +135,7 @@ def main():
135135
ctx = ir.Context()
136136
kernel = create_kernel(ctx)
137137
schedule = create_schedule(ctx)
138+
138139
# Lower the kernel to LLVM dialect.
139140
apply_schedule(kernel, schedule)
140141

0 commit comments

Comments
 (0)