Update _weight_prepack.py (#3424)

DiweiSun · chunyuan-w · web-flow · commit b7b61ea287fb · 2025-01-02T15:33:44.000+08:00
Co-authored-by: Chunyuan WU &lt;chunyuan.wu@intel.com&gt;
diff --git a/intel_extension_for_pytorch/nn/utils/_weight_prepack.py b/intel_extension_for_pytorch/nn/utils/_weight_prepack.py
@@ -59,7 +59,7 @@ def TPPLinear_weight_prepack(m, bk=None, bc=None, layer_dtype=torch.float32):
 #
 # For long term, mark as TODO, we will tune TPP block layout/loop order to make it on par with oneDNN.
 
-fallback_ic_shape_list = [13824, 11008, 16384, 4096]
+fallback_ic_shape_list = [13824, 11008, 16384]
 fallback_oc_shape_list = [4096, 5120, 16384, 12288]
 
 

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ def TPPLinear_weight_prepack(m, bk=None, bc=None, layer_dtype=torch.float32):`
`59`	`59`	`#`
`60`	`60`	`# For long term, mark as TODO, we will tune TPP block layout/loop order to make it on par with oneDNN.`
`61`	`61`
`62`		`-fallback_ic_shape_list = [13824, 11008, 16384, 4096]`
	`62`	`+fallback_ic_shape_list = [13824, 11008, 16384]`
`63`	`63`	`fallback_oc_shape_list = [4096, 5120, 16384, 12288]`
`64`	`64`
`65`	`65`