diff --git a/python/triton/__init__.py b/python/triton/__init__.py
index 3f08b5133..8e118fc00 100644
--- 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