We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ff6ea28 commit df607ebCopy full SHA for df607eb
tests/brevitas_examples/test_llm.py
@@ -417,9 +417,8 @@ def test_small_models_quant_layer(caplog, layer_args):
417
if args.replace_rmsnorm:
418
if torch_version < version.parse('2.4'):
419
pytest.skip("Replacing RMSNorm requires torch 2.4+ or greater")
420
- if hasattr(
421
- args,
422
- 'graph_rotation') and args.graph_rotation == 'fx' and platform.system() == 'win32':
+ if hasattr(args, 'graph_rotation') and args.graph_rotation == 'fx' and platform.system(
+ ) == 'Windows':
423
pytest.skip("Skipping dynamo + windows")
424
float_ppl, quant_ppl, model = validate_args_and_run_main(args)
425
assert_layer_types(model, exp_layer_types)
0 commit comments