fix: fix triton OutOfResources import
This commit is contained in:
parent
50b495f3d8
commit
773aabdda6
|
@ -90,7 +90,7 @@ class Autotuner(triton.KernelInterface):
|
|||
return triton.testing.do_bench(
|
||||
kernel_call, percentiles=(0.5, 0.2, 0.8), rep=40
|
||||
)
|
||||
except triton.compiler.OutOfResources:
|
||||
except triton.OutOfResources:
|
||||
return (float("inf"), float("inf"), float("inf"))
|
||||
|
||||
def run(self, *args, **kwargs):
|
||||
|
|
Loading…
Reference in New Issue