Skip to content

Conversation

masahi
Copy link
Member

@masahi masahi commented Mar 3, 2021

A follow-up to #7425 adding WhileNode support in SPIR-V lowering. It is a simplified from existing ForNode lowering

void CodeGenSPIRV::VisitStmt_(const ForNode* op) {
, removing phi stuff, and loop var init/extent related code. Hopefully it is self explanatory.

Mandelbrot set and binary search test cases from #7425 is working with VK now.

@tqchen @junrushao1994 @vinx13 @jroesch

@masahi masahi merged commit dccc1ab into apache:main Mar 3, 2021
@masahi
Copy link
Member Author

masahi commented Mar 3, 2021

Thnks @tqchen

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
* Add SPIR-V lowering for WhileNode

* test vulkan in while loop tests
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
* Add SPIR-V lowering for WhileNode

* test vulkan in while loop tests
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