Modernize polynomial_custom_function: torch.accelerator and setup_context#3885

jvz37 wants to merge 1 commit intopytorch:mainpytorch/tutorials:mainfrom

jvz37:fix/3880-custom-autograd-modernizejvz37/tutorials:fix/3880-custom-autograd-modernizeCopy head branch name to clipboard