Annotate a few more tests to run in graph mode
PiperOrigin-RevId: 236505757
This commit is contained in:
parent
5ca2639d07
commit
bfa3fcead3
@ -125,6 +125,7 @@ class RecordInputOpTest(test.TestCase):
|
||||
self.assertTrue(r[0] not in epoch_set)
|
||||
epoch_set.add(r[0])
|
||||
|
||||
@test_util.run_deprecated_v1
|
||||
def testDoesNotDeadlock(self):
|
||||
# Iterate multiple times to cause deadlock if there is a chance it can occur
|
||||
for _ in range(30):
|
||||
|
@ -399,6 +399,7 @@ class RFFTOpsTest(BaseFFTOpsTest):
|
||||
rank, (size,) * rank,
|
||||
use_placeholder=True)
|
||||
|
||||
@test_util.run_deprecated_v1
|
||||
def testFftLength(self):
|
||||
if test.is_gpu_available(cuda_only=True):
|
||||
with spectral_ops_test_util.fft_kernel_label_map():
|
||||
|
Loading…
Reference in New Issue
Block a user