Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/127286
Approved by: https://github.com/NicolasHug, https://github.com/DanilBaibak
This commit is contained in:
atalman
2024-05-28 12:38:27 +00:00
committed by PyTorch MergeBot
parent ec8b254ef4
commit a6b994ed54
2 changed files with 3 additions and 1 deletions

View File

@ -7,6 +7,7 @@ import sys
from itertools import product
import numpy as np
import torch
from torch.testing import make_tensor
from torch.testing._internal.common_device_type import (