Do you have a code snippet to reproduce this issue or some specific shape so that we could debug it, please?
As a workaround you could try to disable cudnn for now via torch.backend.cudnn.enabled = False, but I would like to figure out what’s going on.