[PYTHON] Fix-up the previous commit

2021-09-10 11:13:25 -07:00
parent ac10551d55
commit 34369906b4
1 changed files with 1 additions and 1 deletions
--- a/python/triton/init.py
+++ b/python/triton/init.py
@@ -2,7 +2,7 @@
 # or pybind11 shows `munmap_chunk(): invalid pointer`
 import torch
 # submodules
-from .code_gen import cdiv, jit, autotune, heuristics, Config, Autotuner, reinterpret
+from .code_gen import cdiv, next_power_of_2, jit, autotune, heuristics, Config, Autotuner, reinterpret

 from . import language
 from . import code_gen