Skip to content

Commit a14384e

Browse files
committed
[Python][Android] Fix FFI compatibility issues for Android build
1 parent 862a731 commit a14384e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/mlc_llm/support/auto_target.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def build(mod: IRModule, args: "CompileArgs", pipeline=None):
198198
fcompile=tar.tar,
199199
)
200200
if args.debug_dump is not None:
201-
source = ex.mod.imported_modules[0].imported_modules[0].get_source()
201+
source = ex.mod.imports[0].imports[0].inspect_source()
202202
with open(args.debug_dump / "kernel.cl", "w", encoding="utf-8") as f:
203203
f.write(source)
204204

@@ -221,7 +221,7 @@ def build(mod: IRModule, args: "CompileArgs", pipeline=None):
221221
fcompile=ndk.create_shared,
222222
)
223223
if args.debug_dump is not None:
224-
source = ex.mod.imported_modules[0].imported_modules[0].get_source()
224+
source = ex.mod.imports[0].imports[0].inspect_source()
225225
with open(args.debug_dump / "kernel.cl", "w", encoding="utf-8") as f:
226226
f.write(source)
227227

0 commit comments

Comments
 (0)