fixing exp test

HDCharles · HDCharles · commit d73fbd82ba49 · 2025-05-08T11:07:18.000-07:00
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
diff --git a/torchao/experimental/tests/test_int8_dynamic_activation_intx_weight.py b/torchao/experimental/tests/test_int8_dynamic_activation_intx_weight.py
@@ -647,9 +647,9 @@ def test_moe_quant_intx(self):
 
         with torch.device("cpu"):
             model = MOEFeedForwardAOQuantizable(512, 256, 8, 2, empty_init=False).to(
-                torch.bfloat16
+                torch.float32
             )
-            x = torch.randn(8, 512, dtype=torch.bfloat16)
+            x = torch.randn(8, 512, dtype=torch.float32)
 
         out = model(x).clone()
 

Original file line number	Diff line number	Diff line change
`@@ -647,9 +647,9 @@ def test_moe_quant_intx(self):`
`647`	`647`
`648`	`648`	`with torch.device("cpu"):`
`649`	`649`	`model = MOEFeedForwardAOQuantizable(512, 256, 8, 2, empty_init=False).to(`
`650`		`- torch.bfloat16`
	`650`	`+ torch.float32`
`651`	`651`	`)`
`652`		`- x = torch.randn(8, 512, dtype=torch.bfloat16)`
	`652`	`+ x = torch.randn(8, 512, dtype=torch.float32)`
`653`	`653`
`654`	`654`	`out = model(x).clone()`
`655`	`655`