File tree 1 file changed +0
-7
lines changed
1 file changed +0
-7
lines changed Original file line number Diff line number Diff line change @@ -203,51 +203,44 @@ def test_api(self):
203
203
204
204
205
205
class TestPoissonNLLLossFloat16Case (TestPoissonNLLLossBasicCase ):
206
- @test_with_pir_api
207
206
def test_api (self ):
208
207
if core .is_compiled_with_cuda ():
209
208
self .test_static_case (dtype = "float16" )
210
209
self .test_dynamic_case (dtype = "float16" )
211
210
212
211
213
212
class TestPoissonNLLLossBfloat16Case (TestPoissonNLLLossBasicCase ):
214
- @test_with_pir_api
215
213
def test_api (self ):
216
214
if core .is_compiled_with_cuda ():
217
215
self .test_static_case (dtype = "uint16" )
218
216
self .test_dynamic_case (dtype = "uint16" )
219
217
220
218
221
219
class TestPoissonNLLLossFloat32Case (TestPoissonNLLLossBasicCase ):
222
- @test_with_pir_api
223
220
def test_api (self ):
224
221
self .test_static_case (dtype = "float32" )
225
222
self .test_dynamic_case (dtype = "float32" )
226
223
227
224
228
225
class TestPoissonNLLLossFloat64Case (TestPoissonNLLLossBasicCase ):
229
- @test_with_pir_api
230
226
def test_api (self ):
231
227
self .test_static_case (dtype = "float64" )
232
228
self .test_dynamic_case (dtype = "float64" )
233
229
234
230
235
231
class TestPoissonNLLLossNoLoginputCase (TestPoissonNLLLossBasicCase ):
236
- @test_with_pir_api
237
232
def test_api (self ):
238
233
self .test_static_case (log_input = False )
239
234
self .test_dynamic_case (log_input = False )
240
235
241
236
242
237
class TestPoissonNLLLossFulllossCase (TestPoissonNLLLossBasicCase ):
243
- @test_with_pir_api
244
238
def test_api (self ):
245
239
self .test_static_case (full = True )
246
240
self .test_dynamic_case (full = True )
247
241
248
242
249
243
class TestPoissonNLLLossSumReductionCase (TestPoissonNLLLossBasicCase ):
250
- @test_with_pir_api
251
244
def test_api (self ):
252
245
self .test_static_case (reduction = "sum" )
253
246
self .test_dynamic_case (reduction = "sum" )
You can’t perform that action at this time.
0 commit comments