[tests] Test scatter for all backends
@@ -366,7 +366,6 @@ class TestMathFunctions(TestCase):
def test_scatter_1d(self):
for backend in BACKENDS:
- if backend.name in ['NumPy', 'Jax', 'PyTorch']:
with backend:
base = math.ones(x=4)
indices = math.wrap([1, 2], 'points')
@@ -382,7 +381,6 @@ class TestMathFunctions(TestCase):
def test_scatter_update_1d_batched(self):
for backend in BACKENDS:
- if backend.name in ['NumPy', 'Jax', 'PyTorch']:
with backend:
# Only base batched
base = math.zeros(x=4) + math.tensor([0, 1])
@@ -411,7 +409,6 @@ class TestMathFunctions(TestCase):
def test_scatter_update_2d(self):
for backend in BACKENDS:
- if backend.name in ['NumPy', 'Jax', 'PyTorch']:
with backend:
base = math.ones(x=3, y=2)
indices = math.wrap([(0, 0), (0, 1), (2, 1)], 'points,vector')
@@ -421,7 +418,6 @@ class TestMathFunctions(TestCase):
def test_scatter_add_2d(self):
for backend in BACKENDS:
- if backend.name in ('NumPy', 'Jax'):
with backend:
base = math.ones(x=3, y=2)
indices = math.wrap([(0, 0), (0, 0), (0, 1), (2, 1)], 'points,vector')
