Trigger benchamrks' cmake build only when needed

Author: ZzEeKkAa

benchmarks/setup.py

@@ -5,6 +5,7 @@
 import sys
 
 from setuptools import setup
+from setuptools.command.build_ext import build_ext as build_ext_orig
 
 import torch
 
@@ -85,11 +86,18 @@ def build_extension(self):
         subprocess.check_call(["cmake"] + install_args)
 
 
-cmake = CMakeBuild()
-cmake.run()
+class build_ext(build_ext_orig):
+
+    def run(self):
+        cmake = CMakeBuild()
+        cmake.run()
+        super().run()
+
 
 setup(name="triton-kernels-benchmark", packages=[
     "triton_kernels_benchmark",
 ], package_dir={
     "triton_kernels_benchmark": "triton_kernels_benchmark",
-}, package_data={"triton_kernels_benchmark": ["xetla_kernel.cpython-*.so"]})
+}, package_data={"triton_kernels_benchmark": ["xetla_kernel.cpython-*.so"]}, cmdclass={
+    "build_ext": build_ext,
+})