Skip to content

Commit

Permalink
remove device_type from base class
Browse files Browse the repository at this point in the history
  • Loading branch information
Samuel Moors committed Dec 12, 2024
1 parent 88b6df6 commit 42796f3
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion eessi/testsuite/tests/apps/PyTorch/PyTorch_torchvision.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@ class EESSI_PyTorch_torchvision(rfm.RunOnlyRegressionTest, EESSI_Mixin):
nn_model = parameter(['vgg16', 'resnet50', 'resnet152', 'densenet121', 'mobilenet_v3_large'])
bench_name_ci = 'resnet50'
parallel_strategy = parameter([None, 'ddp'])
device_type = variable(str)
# Both torchvision and PyTorch-bundle modules have everything needed to run this test
module_name = parameter(chain(find_modules('torchvision'), find_modules('PyTorch-bundle')))
executable = 'python'
Expand Down

0 comments on commit 42796f3

Please sign in to comment.