[triton-ops] Minor build improvements (#106)

This commit is contained in:
Philippe Tillet
2021-05-12 19:24:11 -04:00
committed by Philippe Tillet
parent d10265f054
commit 147675923e
3 changed files with 2 additions and 14 deletions

View File

@@ -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()

View File

@@ -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",

View File

@@ -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)