Skip to content

[TIR] Support make shape and array at root scope#15709

Closed
wrongtest-intellif wants to merge 1 commit intoapache:mainfrom
wrongtest-intellif:support_make_shape_and_array_in_root_scope
Closed

[TIR] Support make shape and array at root scope#15709
wrongtest-intellif wants to merge 1 commit intoapache:mainfrom
wrongtest-intellif:support_make_shape_and_array_in_root_scope

Conversation

@wrongtest-intellif
Copy link
Copy Markdown
Contributor

@wrongtest-intellif wrongtest-intellif commented Sep 8, 2023

The change wants to allow using stack_make_shape and stack_make_array out of call packed function. A context manager object is introduced to maintain stakc sizes and avoid some duplicate codes.

With the change, if the user has some C function with DLTensor* arguments, it could be invoked through T.call_extern with stack constructed shape and array info.

@wrongtest-intellif
Copy link
Copy Markdown
Contributor Author

@tvm-bot re-run

@tqchen
Copy link
Copy Markdown
Member

tqchen commented Feb 8, 2025

this is useful, sorry we didn't get a chance to merge it in. I worked on the CI and the robustness issue should be resolved, closing for now due to staleness, please raise a new PR if possible. Thank you!

@tqchen tqchen closed this Feb 8, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants