Skip to content

Commit e715814

Browse files
authored
[Relax][Transform] Preserve param names in LiftTransformParams (#16594)
* [Relax][Transform] Preserve param names in LiftTransformParams The `relax.transform.LiftTransformParams` pass splits apart a relax function, extracting the steps that could be performed at compile-time. Prior to this commit, the transformed parameters were named `param0`, `param1`, and so on. This commit updates the `LiftTransformParams` pass to preserve any human-readable parameter names. The parameter names for the updated function are taken from the original parameter names, if no transformation is performed, or from the internal variable binding, if a transformation is applied. This implementation uses `LambdaLift` internally, relying on the changes made in #16306. * Update based on review comments
1 parent b581575 commit e715814

File tree

2 files changed

+380
-343
lines changed

2 files changed

+380
-343
lines changed

0 commit comments

Comments
 (0)