mirrors/mmrazor
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmrazor.git synced 2025-06-03 15:02:54 +08:00
Code Issues Projects Releases Wiki Activity

fix ut

Browse Source
This commit is contained in:
humu789 2023-05-24 15:54:37 +08:00
parent 521696e0aa
commit cb1e4667d9
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_impl/test_quantization/test_gptq/test_op_gptq.py
View File

@ -12,7 +12,8 @@ class TestGPTQOps(unittest.TestCase):
@torch.no_grad()
def test_op(self):
if digit_version(torch.__version__) < digit_version('1.12.0'):
if digit_version(torch.__version__) < digit_version(
'1.12.0') or not torch.cuda.is_available():
self.skipTest('torch<1.12.0')
def get_loss(linear, linear1, data):
@ -59,7 +60,8 @@ class TestGPTQOps(unittest.TestCase):
@torch.no_grad()
def test_model(self):
if digit_version(torch.__version__) < digit_version('1.12.0'):
if digit_version(torch.__version__) < digit_version(
'1.12.0') or not torch.cuda.is_available():
self.skipTest('torch<1.12.0')
import torchvision
model = torchvision.models.resnet18()