If you set torch.backends.cudnn.enabled = False, then it will use the version that I linked above and so if you make your own changes in this code it will run.
Note though that this implementation is not as fast as the one from cudnn as it implements a single algorithm for all cases.