@@ -2924,12 +2924,27 @@ def forward(self, x: Tensor) -> Tensor:
29242924 # mean.default: global mean (dim=None), int32 input → float32 output.
29252925 # torch promotes the operand to the requested dtype BEFORE averaging.
29262926 (torch .randint (0 , 10 , (3 , 4 ), dtype = torch .int32 ), None , False , torch .float32 ),
2927- (torch .randint (- 5 , 5 , (2 , 3 , 4 ), dtype = torch .int32 ), None , False , torch .float32 ),
2927+ (
2928+ torch .randint (- 5 , 5 , (2 , 3 , 4 ), dtype = torch .int32 ),
2929+ None ,
2930+ False ,
2931+ torch .float32 ,
2932+ ),
29282933 # mean.dim: reduce along specific dimensions, int32 input → float32.
29292934 (torch .randint (0 , 10 , (3 , 4 ), dtype = torch .int32 ), 1 , False , torch .float32 ),
29302935 (torch .randint (0 , 10 , (3 , 4 ), dtype = torch .int32 ), 0 , True , torch .float32 ),
2931- (torch .randint (- 5 , 5 , (2 , 3 , 4 ), dtype = torch .int32 ), [0 , 2 ], False , torch .float32 ),
2932- (torch .randint (- 5 , 5 , (2 , 3 , 4 ), dtype = torch .int32 ), [0 , 2 ], True , torch .float32 ),
2936+ (
2937+ torch .randint (- 5 , 5 , (2 , 3 , 4 ), dtype = torch .int32 ),
2938+ [0 , 2 ],
2939+ False ,
2940+ torch .float32 ,
2941+ ),
2942+ (
2943+ torch .randint (- 5 , 5 , (2 , 3 , 4 ), dtype = torch .int32 ),
2944+ [0 , 2 ],
2945+ True ,
2946+ torch .float32 ,
2947+ ),
29332948 # dtype-agnostic: root cause is independent of the specific int operand
29342949 # dtype or the specific float target dtype.
29352950 (torch .randint (0 , 10 , (3 , 4 ), dtype = torch .int64 ), 1 , False , torch .float32 ),
0 commit comments