[Dev] Correcting cxx compiler (#294)

tilelang/contrib/cc.py

@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Util to invoke C/C++ compilers in the system."""
+import functools
 import os
 import shutil
 import subprocess
@@ -62,6 +63,7 @@ def get_cc():
     return None
 
 
+@functools.lru_cache(maxsize=None)
 def get_cplus_compiler():
     """Return the path to the default C/C++ compiler.
 

tilelang/jit/adapter/libgen.py

@@ -67,10 +67,11 @@ class LibraryGenerator(object):
                 src.name,
             ]
         elif is_cpu_target(target):
+            from tilelang.contrib.cc import get_cplus_compiler
             src = tempfile.NamedTemporaryFile(mode="w", suffix=".cpp", delete=False)
             libpath = src.name.replace(".cpp", ".so")
 
-            command = ["g++", "-std=c++17", "-fPIC", "-shared", src.name]
+            command = [get_cplus_compiler(), "-std=c++17", "-fPIC", "-shared", src.name]
             with_tl = False
             command += [
                 "-I" + TILELANG_TEMPLATE_PATH,