Arm backend: Update weight observer for QAT

zingo · zingo · commit e1fa3319a91e · 2025-09-09T16:20:48.000+02:00
Updates weight observer for QAT+per-channel. Also adds a simple test to
test QAT flow.

Signed-off-by: Zingo Andersen &lt;zingo.andersen@arm.com&gt;
Change-Id: I69531add22ae4c7a3912916d6fc46fde0217e8df
diff --git a/backends/arm/test/misc/test_qat_training_loop.py b/backends/arm/test/misc/test_qat_training_loop.py
@@ -11,7 +11,7 @@
     TOSAQuantizer,
 )
 
-from executorch.backends.arm.tosa_specification import TosaSpecification
+from executorch.backends.arm.tosa.specification import TosaSpecification
 from torch.export import export
 from torchao.quantization.pt2e import (
     move_exported_model_to_eval,

Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,7 @@`
`11`	`11`	`TOSAQuantizer,`
`12`	`12`	`)`
`13`	`13`
`14`		`-from executorch.backends.arm.tosa_specification import TosaSpecification`
	`14`	`+from executorch.backends.arm.tosa.specification import TosaSpecification`
`15`	`15`	`from torch.export import export`
`16`	`16`	`from torchao.quantization.pt2e import (`
`17`	`17`	`move_exported_model_to_eval,`