diff --git a/CMakeLists.txt b/CMakeLists.txt index 5f65560c4..23887c91e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -92,9 +92,3 @@ target_link_libraries(triton ${LLVM_LIBRARIES}) if(BUILD_PYTHON_MODULE) target_link_libraries(triton ${TORCH_LIBRARIES} ${CUTLASS_LIBRARIES}) endif() - -# Tutorials -if(BUILD_TUTORIALS) - message(STATUS "Adding C++ tutorials") - add_subdirectory(tutorials) -endif() diff --git a/python/setup.py b/python/setup.py index 636054b0e..ed67e3aa6 100644 --- a/python/setup.py +++ b/python/setup.py @@ -80,8 +80,9 @@ class CMakeBuild(build_ext): cmake_args += ["-A", "x64"] build_args += ["--", "/m"] else: + import multiprocessing cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg] - build_args += ["--", "-j8"] + build_args += ["--", f'-j{2*multiprocessing.cpu_count()}'] env = os.environ.copy() subprocess.check_call(["cmake", self.base_dir] + cmake_args, cwd=self.build_temp, env=env) @@ -105,7 +106,6 @@ setup( # for PyPI keywords=["Compiler", "Deep Learning"], url="https://github.com/ptillet/triton/", - download_url="https://github.com/ptillet/triton/archive/v0.1.tar.gz", classifiers=[ "Development Status :: 4 - Beta", "Intended Audience :: Developers", diff --git a/tutorials/CMakeLists.txt b/tutorials/CMakeLists.txt deleted file mode 100644 index 1dd6c9b8f..000000000 --- a/tutorials/CMakeLists.txt +++ /dev/null @@ -1,6 +0,0 @@ -foreach(PROG 01-matmul) - set(TARGET ${PROG}) - add_executable(${TARGET} ${PROG}.cc) - set_target_properties(${TARGET} PROPERTIES OUTPUT_NAME ${TARGET}) - target_link_libraries(${TARGET} triton dl) -endforeach(PROG)