mplxrg
mplxrg
Sorry for the late reply. I made a simple test case as follows. In this example, I split the parameters into two arrays and explicitly specified which one requires grad...
@tgymnich Thank you so much for your answer! I'm trying to detach a intermediate variable inside a very complex function not a parameter of the function. I try to use...
Option B is exactly what I'm seeking. `__enzyme_inactivefn` seems to be a new feature in v0.0.17 which works pretty well for my use case. One thing to note here is...
That would be great! `__enzyme_inactiveval` sounds good to me. Another name I could come up with is `__enzyme_detach()` which is similar to torch.detach().