Commit e715814
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 comments1 parent b581575 commit e715814
File tree
2 files changed
+380
-343
lines changed- src/relax/transform
- tests/python/relax
2 files changed
+380
-343
lines changed
0 commit comments