Remove Total Time Measurement (#1105)

Remove total time measurement
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

setup.py

@@ -41,9 +41,6 @@ elif "jax" in frameworks:
     install_and_import("pybind11[global]")
     from pybind11.setup_helpers import build_ext as BuildExtension
 
-# Start timing
-start_time = time.perf_counter()
-
 
 CMakeBuildExtension = get_build_ext(BuildExtension)
 
@@ -55,7 +52,7 @@ class TimedBdist(bdist_wheel):
         start_time = time.perf_counter()
         super().run()
         total_time = time.perf_counter() - start_time
-        print(f"Time for bdist_wheel: {total_time:.2f} seconds")
+        print(f"Total time for bdist_wheel: {total_time:.2f} seconds")
 
 
 def setup_common_extension() -> CMakeExtension:
@@ -171,8 +168,3 @@ if __name__ == "__main__":
         include_package_data=True,
         package_data={"": ["VERSION.txt"]},
     )
-
-    # End timing
-    end_time = time.perf_counter()
-    total_time = end_time - start_time
-    print(f"Total build time: {total_time:.2f} seconds")