Skip to content

Commit 8dd8692

Browse files
tp5uiucclaude
andcommitted
test(rtx): remove remaining BF16 skip guards on TensorRT-RTX
BF16 is now supported on TensorRT-RTX via the per-layer depthwise fallback validator. Remove the leftover skipIf guards that were blanket-disabling BF16 tests on RTX. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent d930aa9 commit 8dd8692

4 files changed

Lines changed: 0 additions & 20 deletions

File tree

tests/py/dynamo/conversion/test_binary_ops_aten.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -237,10 +237,6 @@ def forward(self, x, y):
237237
if op[0].__name__ not in ["pow.Tensor_Tensor", "fmod.Tensor"]
238238
]
239239
)
240-
@unittest.skipIf(
241-
torch_tensorrt.ENABLED_FEATURES.tensorrt_rtx,
242-
"bf16 is not supported for tensorrt_rtx",
243-
)
244240
def test_elementwise_ops_bf16(self, _, orig_op):
245241
class TestModule(nn.Module):
246242
def __init__(self, orig_op):

tests/py/dynamo/conversion/test_casts.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -67,10 +67,6 @@ def forward(self, x):
6767
precision=torch.float,
6868
)
6969

70-
@unittest.skipIf(
71-
torch_tensorrt.ENABLED_FEATURES.tensorrt_rtx,
72-
"bf16 is not supported for tensorrt_rtx",
73-
)
7470
def test_to_copy_bfloat16(self):
7571
class ToCopyBFloat16(nn.Module):
7672
def forward(self, x):

tests/py/dynamo/models/test_dtype_support.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -200,10 +200,6 @@ def forward(self, x):
200200
),
201201
"Platform does not have BF16 support",
202202
)
203-
@unittest.skipIf(
204-
torch_tensorrt.ENABLED_FEATURES.tensorrt_rtx,
205-
"bf16 is not supported for tensorrt_rtx",
206-
)
207203
class TestBF16Support(TestCase):
208204
@unittest.skipIf(
209205
not torch_tensorrt.ENABLED_FEATURES.torch_tensorrt_runtime,

tests/py/dynamo/models/test_models.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -519,10 +519,6 @@ def forward(
519519

520520

521521
@pytest.mark.unit
522-
@unittest.skipIf(
523-
torchtrt.ENABLED_FEATURES.tensorrt_rtx,
524-
"bf16 is not supported for tensorrt_rtx",
525-
)
526522
@pytest.mark.critical
527523
def test_bf16_model(ir):
528524
class MyModule(torch.nn.Module):
@@ -568,10 +564,6 @@ def forward(self, x):
568564

569565

570566
@pytest.mark.unit
571-
@unittest.skipIf(
572-
torchtrt.ENABLED_FEATURES.tensorrt_rtx,
573-
"bf16 is not supported for tensorrt_rtx",
574-
)
575567
@pytest.mark.critical
576568
def test_bf16_fallback_model(ir):
577569
class MyModule(torch.nn.Module):

0 commit comments

Comments
 (0)