Skip to content

Commit db4ba94

Browse files
authored
[PIR]Open uts for 281-295 (#60894)
1 parent dd79085 commit db4ba94

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

test/legacy_test/test_poisson_nll_loss.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -203,51 +203,44 @@ def test_api(self):
203203

204204

205205
class TestPoissonNLLLossFloat16Case(TestPoissonNLLLossBasicCase):
206-
@test_with_pir_api
207206
def test_api(self):
208207
if core.is_compiled_with_cuda():
209208
self.test_static_case(dtype="float16")
210209
self.test_dynamic_case(dtype="float16")
211210

212211

213212
class TestPoissonNLLLossBfloat16Case(TestPoissonNLLLossBasicCase):
214-
@test_with_pir_api
215213
def test_api(self):
216214
if core.is_compiled_with_cuda():
217215
self.test_static_case(dtype="uint16")
218216
self.test_dynamic_case(dtype="uint16")
219217

220218

221219
class TestPoissonNLLLossFloat32Case(TestPoissonNLLLossBasicCase):
222-
@test_with_pir_api
223220
def test_api(self):
224221
self.test_static_case(dtype="float32")
225222
self.test_dynamic_case(dtype="float32")
226223

227224

228225
class TestPoissonNLLLossFloat64Case(TestPoissonNLLLossBasicCase):
229-
@test_with_pir_api
230226
def test_api(self):
231227
self.test_static_case(dtype="float64")
232228
self.test_dynamic_case(dtype="float64")
233229

234230

235231
class TestPoissonNLLLossNoLoginputCase(TestPoissonNLLLossBasicCase):
236-
@test_with_pir_api
237232
def test_api(self):
238233
self.test_static_case(log_input=False)
239234
self.test_dynamic_case(log_input=False)
240235

241236

242237
class TestPoissonNLLLossFulllossCase(TestPoissonNLLLossBasicCase):
243-
@test_with_pir_api
244238
def test_api(self):
245239
self.test_static_case(full=True)
246240
self.test_dynamic_case(full=True)
247241

248242

249243
class TestPoissonNLLLossSumReductionCase(TestPoissonNLLLossBasicCase):
250-
@test_with_pir_api
251244
def test_api(self):
252245
self.test_static_case(reduction="sum")
253246
self.test_dynamic_case(reduction="sum")

0 commit comments

Comments
 (0)