Skip to content

Commit 5fe8a92

Browse files
committed
Let's see if these eleven pass now.
1 parent 472a5ff commit 5fe8a92

File tree

1 file changed

+0
-11
lines changed

1 file changed

+0
-11
lines changed

test/test_sparse.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,6 @@ def randn(self, *args, **kwargs):
107107
# TODO: Put this in torch.cuda.randn
108108
return self.ValueTensor(*args, **kwargs).normal_()
109109

110-
@skipIfRocm
111110
def test_basic(self):
112111
x, i, v = self._gen_sparse(3, 10, 100)
113112

@@ -156,7 +155,6 @@ def test_ctor_size_checks(self):
156155
RuntimeError,
157156
lambda: self.SparseTensor(indices, values, torch.Size([2, 4, 2, 1])))
158157

159-
@skipIfRocm
160158
def test_to_dense(self):
161159
i = self.IndexTensor([
162160
[0, 1, 2, 2],
@@ -186,7 +184,6 @@ def test_to_dense(self):
186184
self.assertEqual(res, x.to_dense())
187185
self.assertEqual(res, self.safeToDense(x))
188186

189-
@skipIfRocm
190187
def test_shared(self):
191188
i = self.IndexTensor([[2]])
192189
v = self.ValueTensor([5])
@@ -196,7 +193,6 @@ def test_shared(self):
196193
i[0][0] = 0
197194
self.assertEqual(self.ValueTensor([6, 0, 0]), self.safeToDense(x))
198195

199-
@skipIfRocm
200196
def test_to_dense_hybrid(self):
201197
i = self.IndexTensor([
202198
[0, 1, 2, 2],
@@ -225,7 +221,6 @@ def test_to_dense_hybrid(self):
225221
self.assertEqual(res, x.to_dense())
226222
self.assertEqual(res, self.safeToDense(x))
227223

228-
@skipIfRocm
229224
def test_contig(self):
230225
i = self.IndexTensor([
231226
[1, 0, 35, 14, 39, 6, 71, 66, 40, 27],
@@ -279,7 +274,6 @@ def test_contig(self):
279274
self.assertEqual(exp_i, x._indices())
280275
self.assertEqual(exp_v, x._values())
281276

282-
@skipIfRocm
283277
def test_contig_hybrid(self):
284278
i = self.IndexTensor([
285279
[1, 0, 35, 14, 39, 6, 71, 66, 40, 27],
@@ -339,7 +333,6 @@ def test_contig_hybrid(self):
339333
self.assertEqual(exp_i, x._indices())
340334
self.assertEqual(exp_v, x._values())
341335

342-
@skipIfRocm
343336
def test_clone(self):
344337
x, _, _ = self._gen_sparse(4, 20, 5)
345338
if self.is_uncoalesced:
@@ -361,7 +354,6 @@ def test_cuda_empty(self):
361354
self.assertEqual(y._sparseDims(), x._sparseDims())
362355
self.assertEqual(y._denseDims(), x._denseDims())
363356

364-
@skipIfRocm
365357
def test_transpose(self):
366358
x = self._gen_sparse(4, 20, 5)[0]
367359
y = self.safeToDense(x)
@@ -410,7 +402,6 @@ def test_t_empty(self):
410402
self.assertEqual(x._sparseDims(), 2)
411403
self.assertEqual(x._denseDims(), 0)
412404

413-
@skipIfRocm
414405
def test_add_zeros(self):
415406
def test_shape(sparse_dims, sizes):
416407
x, _, _ = self._gen_sparse(sparse_dims, 20, sizes)
@@ -474,7 +465,6 @@ def test_shape(di, dj, dk):
474465
test_shape(1000, 100, 100)
475466
test_shape(3000, 64, 300)
476467

477-
@skipIfRocm
478468
def test_dsmm(self):
479469
def test_shape(di, dj, dk):
480470
x = self._gen_sparse(2, 20, [di, dj])[0]
@@ -488,7 +478,6 @@ def test_shape(di, dj, dk):
488478
test_shape(1000, 100, 100)
489479
test_shape(3000, 64, 300)
490480

491-
@skipIfRocm
492481
def test_hsmm(self):
493482
def test_shape(di, dj, dk):
494483
x = self._gen_sparse(2, 20, [di, dj])[0]

0 commit comments

Comments
 (0)