@altanh
That’s not it. The output doesn’t change if I remove the dead code elimination.
And I just rebuilt tvm, the issue is still there on the current master.
(Had to add a mod = relay.transform.InferType()(mod) before the gradient though)
1 Like