@@ -37,15 +37,15 @@ def fp32_fallback_lowering(x: object) -> object:
37
37
38
38
# Operations that need fp32 fallbacks due to libdevice/tl_math limitations
39
39
FP32_FALLBACK_OPS_UNARY = [
40
- torch .ops .aten .rsqrt .default ,
41
- torch .ops .aten .sqrt .default ,
42
- torch .ops .aten .sin .default ,
43
- torch .ops .aten .cos .default ,
44
- torch .ops .aten .log .default ,
45
- torch .ops .aten .tanh .default ,
46
- torch .ops .aten .log1p .default ,
47
- torch .ops .aten .expm1 .default ,
48
- torch .ops .aten .exp .default ,
40
+ torch .ops .aten .rsqrt .default , # pyright: ignore[reportAttributeAccessIssue]
41
+ torch .ops .aten .sqrt .default , # pyright: ignore[reportAttributeAccessIssue]
42
+ torch .ops .aten .sin .default , # pyright: ignore[reportAttributeAccessIssue]
43
+ torch .ops .aten .cos .default , # pyright: ignore[reportAttributeAccessIssue]
44
+ torch .ops .aten .log .default , # pyright: ignore[reportAttributeAccessIssue]
45
+ torch .ops .aten .tanh .default , # pyright: ignore[reportAttributeAccessIssue]
46
+ torch .ops .aten .log1p .default , # pyright: ignore[reportAttributeAccessIssue]
47
+ torch .ops .aten .expm1 .default , # pyright: ignore[reportAttributeAccessIssue]
48
+ torch .ops .aten .exp .default , # pyright: ignore[reportAttributeAccessIssue]
49
49
]
50
50
51
51
# Register fp32 fallback lowerings for ops that don't support fp16/bfloat16
0 commit comments