[REFACTOR][IR] Generalize staging lambdas and explicit thread reductions - #20552

[REFACTOR][IR] Generalize staging lambdas and explicit thread reductions#20552

tqchen merged 7 commits into

apache:mainapache/tvm:mainfrom

tqchen:ir/staging-lambda-allreducetqchen/tvm:ir/staging-lambda-allreduceCopy head branch name to clipboard