Hi,
While using amp with my custom forward function using autograd.function. I am getting below error:
NameError: name ‘custom_fwd’ is not defined
I have added @custom_fwd decorator above my forward function.