File renaming.

vbharadwaj-bk · vbharadwaj-bk · commit 596874576f4b · 2026-03-22T14:12:15.000-07:00
diff --git a/openequivariance/openequivariance/_torch/CUETensorProduct.py b/openequivariance/openequivariance/_torch/CUETensorProduct.py
@@ -6,7 +6,7 @@
 
 from openequivariance.core.TensorProductBase import TensorProductBase
 from openequivariance.core.e3nn_lite import TPProblem
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance.benchmark.tpp_creation_utils import (
     ChannelwiseTPP,
     FullyConnectedTPProblem,
diff --git a/openequivariance/openequivariance/_torch/E3NNTensorProduct.py b/openequivariance/openequivariance/_torch/E3NNTensorProduct.py
@@ -11,7 +11,7 @@
 
 from openequivariance.core.TensorProductBase import TensorProductBase
 from openequivariance.core.e3nn_lite import TPProblem
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance._torch.NPDoubleBackwardMixin import NumpyDoubleBackwardMixin
 
 TORCH_COMPILE_AUTOTUNING_DIR = pathlib.Path("triton_autotuning")
diff --git a/openequivariance/openequivariance/_torch/TensorProduct.py b/openequivariance/openequivariance/_torch/TensorProduct.py
@@ -3,7 +3,7 @@
 from openequivariance._torch import extlib
 import torch
 from openequivariance.core.utils import torch_to_oeq_dtype, dtype_to_enum
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance._torch.utils import (
     reorder_torch,
     string_to_tensor,
diff --git a/openequivariance/openequivariance/_torch/TensorProductConv.py b/openequivariance/openequivariance/_torch/TensorProductConv.py
@@ -23,7 +23,7 @@
     enum_to_torch_dtype,
 )
 
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance._torch.NPDoubleBackwardMixin import NumpyDoubleBackwardMixinConv
 
 logger = getLogger()
diff --git a/openequivariance/openequivariance/_torch/extlib/__init__.py b/openequivariance/openequivariance/_torch/extlib/__init__.py
@@ -8,7 +8,7 @@
 
 import torch
 
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 
 oeq_root = str(Path(__file__).parent.parent.parent)
 
diff --git a/openequivariance/openequivariance/benchmark/ConvBenchmarkSuite.py b/openequivariance/openequivariance/benchmark/ConvBenchmarkSuite.py
@@ -6,7 +6,7 @@
 import numpy as np
 
 import openequivariance as oeq
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance.core.ConvolutionBase import CoordGraph
 from openequivariance.benchmark.benchmark_utils import NpEncoder
 
diff --git a/openequivariance/openequivariance/benchmark/TestBenchmarkSuite.py b/openequivariance/openequivariance/benchmark/TestBenchmarkSuite.py
@@ -10,9 +10,9 @@
 from openequivariance._torch.extlib import DeviceProp
 from openequivariance.core.TensorProductBase import TensorProductBase
 
-from openequivariance.benchmark.logging_utils import getLogger, bcolors
+from openequivariance.benchmark.logging import getLogger, bcolors
 from openequivariance.core.e3nn_lite import TPProblem
-from openequivariance.benchmark.correctness_utils import (
+from openequivariance.benchmark.correctness import (
     correctness_forward,
     correctness_backward,
     correctness_double_backward,
diff --git a/openequivariance/openequivariance/benchmark/benchmark_utils.py b/openequivariance/openequivariance/benchmark/benchmark_utils.py
@@ -15,7 +15,7 @@
 from openequivariance.core.TensorProductBase import TensorProductBase
 from openequivariance.core.e3nn_lite import TPProblem
 from openequivariance._torch.CUETensorProduct import CUETensorProduct
-from openequivariance.benchmark.logging_utils import getLogger, bcolors
+from openequivariance.benchmark.logging import getLogger, bcolors
 
 logger = getLogger()
 
diff --git a/openequivariance/openequivariance/benchmark/correctness.py b/openequivariance/openequivariance/benchmark/correctness.py
@@ -4,7 +4,7 @@
 import numpy.linalg as la
 
 from openequivariance._torch.CUETensorProduct import CUETensorProduct
-from openequivariance.benchmark.logging_utils import bcolors, getLogger
+from openequivariance.benchmark.logging import bcolors, getLogger
 from openequivariance.benchmark.random_buffer_utils import (
     get_random_buffers_backward,
     get_random_buffers_double_backward,
diff --git a/openequivariance/openequivariance/benchmark/logging.py b/openequivariance/openequivariance/benchmark/logging.py
diff --git a/openequivariance/openequivariance/benchmark/perf_metrics_utils.py b/openequivariance/openequivariance/benchmark/perf_metrics_utils.py
@@ -6,7 +6,7 @@
 )
 
 from openequivariance.core.e3nn_lite import TPProblem, wigner_3j
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 import numpy as np
 
 logger = getLogger()
diff --git a/openequivariance/openequivariance/core/ComputationSchedule.py b/openequivariance/openequivariance/core/ComputationSchedule.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance.core.e3nn_lite import Irreps, TPProblem, wigner_3j
 
 logger = getLogger()
diff --git a/openequivariance/openequivariance/core/ConvolutionBase.py b/openequivariance/openequivariance/core/ConvolutionBase.py
@@ -2,8 +2,8 @@
 
 import numpy as np
 
-from openequivariance.benchmark.correctness_utils import check_similiarity
-from openequivariance.benchmark.logging_utils import bcolors, getLogger
+from openequivariance.benchmark.correctness import check_similiarity
+from openequivariance.benchmark.logging import bcolors, getLogger
 from openequivariance.benchmark.random_buffer_utils import (
     get_random_buffers_backward_conv,
     get_random_buffers_double_backward_conv,
diff --git a/openequivariance/openequivariance/core/LoopUnrollTP.py b/openequivariance/openequivariance/core/LoopUnrollTP.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance.core.ComputationSchedule import (
     ComputationSchedule,
     SMEMCapacityException,
diff --git a/openequivariance/openequivariance/core/TensorProductBase.py b/openequivariance/openequivariance/core/TensorProductBase.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 from openequivariance.core.e3nn_lite import TPProblem
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance.core.utils import benchmark
 
 logger = getLogger()
diff --git a/openequivariance/openequivariance/jax/TensorProductConv.py b/openequivariance/openequivariance/jax/TensorProductConv.py
@@ -9,7 +9,7 @@
 from openequivariance.core.LoopUnrollConv import LoopUnrollConv
 from openequivariance.jax.utils import reorder_jax
 
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance.jax.jvp import conv_prim
 from openequivariance.jax.vjp import conv_func
 
diff --git a/tests/batch_test.py b/tests/batch_test.py
@@ -3,7 +3,7 @@
 import numpy as np
 import pytest
 import torch
-from openequivariance.benchmark.correctness_utils import (
+from openequivariance.benchmark.correctness import (
     correctness_backward,
     correctness_double_backward,
     correctness_forward,
diff --git a/tests/benchmark.py b/tests/benchmark.py
@@ -10,7 +10,7 @@
 
 import numpy as np
 
-from openequivariance.benchmark.logging_utils import getLogger
+from openequivariance.benchmark.logging import getLogger
 from openequivariance._torch.extlib import DeviceProp
 from openequivariance._torch.E3NNTensorProduct import (
     E3NNTensorProduct,

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@`
`23`	`23`	`enum_to_torch_dtype,`
`24`	`24`	`)`
`25`	`25`
`26`		`-from openequivariance.benchmark.logging_utils import getLogger`
	`26`	`+from openequivariance.benchmark.logging import getLogger`
`27`	`27`	`from openequivariance._torch.NPDoubleBackwardMixin import NumpyDoubleBackwardMixinConv`
`28`	`28`
`29`	`29`	`logger = getLogger()`