File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -119,6 +119,9 @@ def read_log(log_dir: str) -> str:
119119
120120
121121def test_sanity (feature_dirs ):
122+ if not fp8_available :
123+ pytest .skip (reason_for_no_fp8 )
124+
122125 log_all_stats_config = LOG_QUANTIZED_CONFIG_BASE .format (stats = ", " .join (all_stats ))
123126 with debug_session (log_all_stats_config , feature_dirs ) as log_dir :
124127 model = te .Linear (128 , 128 , params_dtype = torch .bfloat16 )
@@ -207,6 +210,9 @@ def test_numerics(fp8_recipe, feature_dirs):
207210
208211@pytest .mark .parametrize ("layer" , ["linear" , "transformer" ])
209212def test_log_every_3_or_5_layers (layer , configs_dir , feature_dirs ):
213+ if not fp8_available :
214+ pytest .skip (reason_for_no_fp8 )
215+
210216 # If layer does not invoke any feature in current iteration,
211217 # then it changed into non-debug mode.
212218 # This test checks whether this works correctly -
You can’t perform that action at this time.
0 commit comments