Skip to content

Commit d92ddcf

Browse files
bddppqfacebook-github-bot
authored andcommitted
Skip convnets benchmark in rocm CI (pytorch#17331)
Summary: random coredump Pull Request resolved: pytorch#17331 Differential Revision: D14162018 Pulled By: bddppq fbshipit-source-id: 3ed15a79b7bca2498c50f6af80cbd6be7229dea8
1 parent b3b692a commit d92ddcf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

caffe2/python/convnet_benchmarks_test.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
from caffe2.python import test_util, workspace
44

55

6-
@unittest.skipIf(not workspace.has_gpu_support, "no gpu")
6+
# TODO: investigate why this randomly core dump in ROCM CI
7+
@unittest.skipIf(not workspace.has_cuda_support, "no cuda gpu")
78
class TestConvnetBenchmarks(test_util.TestCase):
89
def testConvnetBenchmarks(self):
910
all_args = [

0 commit comments

Comments
 (0)