Move the ufunc module

cunumeric/__init__.py

@@ -28,11 +28,7 @@
 from cunumeric import linalg, random
 from cunumeric.array import ndarray
 from cunumeric.module import *
-from cunumeric.ufunc.bit_twiddling import *
-from cunumeric.ufunc.comparison import *
-from cunumeric.ufunc.math import *
-from cunumeric.ufunc.floating import *
-from cunumeric.ufunc.trigonometric import *
+from cunumeric._ufunc import *
 from cunumeric.coverage import clone_module
 
 clone_module(_np, globals())

cunumeric/ufunc/__init__.py → cunumeric/_ufunc/__init__.py

@@ -12,3 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+
+from .bit_twiddling import *
+from .comparison import *
+from .math import *
+from .floating import *
+from .trigonometric import *
+from .ufunc import ufunc

cunumeric/linalg/linalg.py

@@ -14,9 +14,9 @@
 #
 
 import numpy as np
+from cunumeric._ufunc.math import sqrt as _sqrt
 from cunumeric.array import convert_to_cunumeric_ndarray
 from cunumeric.module import ndarray
-from cunumeric.ufunc.math import sqrt as _sqrt
 
 
 def cholesky(a):

cunumeric/module.py

@@ -23,9 +23,9 @@
 
 import numpy as np
 import opt_einsum as oe
-from cunumeric.ufunc.comparison import maximum, minimum
-from cunumeric.ufunc.floating import floor
-from cunumeric.ufunc.math import add, multiply
+from cunumeric._ufunc.comparison import maximum, minimum
+from cunumeric._ufunc.floating import floor
+from cunumeric._ufunc.math import add, multiply
 
 from .array import (
     convert_to_cunumeric_ndarray,

setup.py

@@ -79,7 +79,7 @@ def run(self):
             "cunumeric",
             "cunumeric.linalg",
             "cunumeric.random",
-            "cunumeric.ufunc",
+            "cunumeric._ufunc",
         ],
         cmdclass={"build_py": my_build_py},
     )