Fix hybrid fp8 test guard (#3492)

Signed-off-by: Evgeny <etsykunov@nvidia.com>

Author: negvet

tests/pytorch/distributed/fsdp2_tests/run_fsdp2_fused_adam.py

@@ -1733,6 +1733,10 @@ def test_fused_adam_hybrid_scale_uniform_across_shards(hybrid_recipe_name):
     ), f"missing hybrid current-scaling directions: {checked}"
 
 
+@pytest.mark.skipif(
+    not te.is_fp8_available(),
+    reason=te.is_fp8_available(return_reason=True)[1],
+)
 def test_fused_adam_hybrid_identity_fp8_master_weights():
     """FSDP2 + FusedAdam with Hybrid(FP8 current rowwise, Identity columnwise).
 

tests/pytorch/distributed/fsdp2_tests/run_fsdp2_model.py

@@ -510,6 +510,10 @@ def _hybrid_param_count():
     _check_fp8_fsdp2_allgather(model, tols=dict(atol=5e-4, rtol=5e-3))
 
 
+@pytest.mark.skipif(
+    not te.is_fp8_available(),
+    reason=te.is_fp8_available(return_reason=True)[1],
+)
 def test_distributed_hybrid_identity_all():
     """FSDP2 training/all-gather with an all-Identity CustomRecipe.