From 34369906b4637ea6d4fa57707cd93dd3be91699d Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Fri, 10 Sep 2021 11:13:25 -0700 Subject: [PATCH] [PYTHON] Fix-up the previous commit --- python/triton/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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