Skip to content

Commit 69703c2

Browse files
authored
Fix test failures (#218)
1 parent 856dadf commit 69703c2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

torchinductor/codegen/triton.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def constant(value, dtype):
5656
return 'float("inf")'
5757
elif value == float("-inf"):
5858
return 'float("-inf")'
59-
return super().constant(value, dtype)
59+
return OpOverrides.constant(value, dtype)
6060

6161
@staticmethod
6262
def abs(x):

torchinductor/decomposition.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def rsub(a, b):
140140
return b - a
141141

142142

143-
@register_decomposition([aten.pow.Tensor_Scalar], decompositions)
143+
@register_decomposition([aten.pow], decompositions)
144144
def pow(a, b):
145145
# see https://github.com/openai/triton/issues/506
146146
# triton doesn't support pow, so need to rewrite it

0 commit comments

Comments
 (0)