Skip to content

Commit 660732c

Browse files
authored
Fix pyright errors (#370)
1 parent 7f1ef4e commit 660732c

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

helion/_compiler/inductor_lowering_extra.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,15 @@ def fp32_fallback_lowering(x: object) -> object:
3737

3838
# Operations that need fp32 fallbacks due to libdevice/tl_math limitations
3939
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]
4949
]
5050

5151
# Register fp32 fallback lowerings for ops that don't support fp16/bfloat16

0 commit comments

Comments
 (0)