[triton-ops] Minor build improvements (#106)
This commit is contained in:
committed by
Philippe Tillet
parent
d10265f054
commit
147675923e
@@ -92,9 +92,3 @@ target_link_libraries(triton ${LLVM_LIBRARIES})
|
|||||||
if(BUILD_PYTHON_MODULE)
|
if(BUILD_PYTHON_MODULE)
|
||||||
target_link_libraries(triton ${TORCH_LIBRARIES} ${CUTLASS_LIBRARIES})
|
target_link_libraries(triton ${TORCH_LIBRARIES} ${CUTLASS_LIBRARIES})
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
# Tutorials
|
|
||||||
if(BUILD_TUTORIALS)
|
|
||||||
message(STATUS "Adding C++ tutorials")
|
|
||||||
add_subdirectory(tutorials)
|
|
||||||
endif()
|
|
||||||
|
@@ -80,8 +80,9 @@ class CMakeBuild(build_ext):
|
|||||||
cmake_args += ["-A", "x64"]
|
cmake_args += ["-A", "x64"]
|
||||||
build_args += ["--", "/m"]
|
build_args += ["--", "/m"]
|
||||||
else:
|
else:
|
||||||
|
import multiprocessing
|
||||||
cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg]
|
cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg]
|
||||||
build_args += ["--", "-j8"]
|
build_args += ["--", f'-j{2*multiprocessing.cpu_count()}']
|
||||||
|
|
||||||
env = os.environ.copy()
|
env = os.environ.copy()
|
||||||
subprocess.check_call(["cmake", self.base_dir] + cmake_args, cwd=self.build_temp, env=env)
|
subprocess.check_call(["cmake", self.base_dir] + cmake_args, cwd=self.build_temp, env=env)
|
||||||
@@ -105,7 +106,6 @@ setup(
|
|||||||
# for PyPI
|
# for PyPI
|
||||||
keywords=["Compiler", "Deep Learning"],
|
keywords=["Compiler", "Deep Learning"],
|
||||||
url="https://github.com/ptillet/triton/",
|
url="https://github.com/ptillet/triton/",
|
||||||
download_url="https://github.com/ptillet/triton/archive/v0.1.tar.gz",
|
|
||||||
classifiers=[
|
classifiers=[
|
||||||
"Development Status :: 4 - Beta",
|
"Development Status :: 4 - Beta",
|
||||||
"Intended Audience :: Developers",
|
"Intended Audience :: Developers",
|
||||||
|
@@ -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)
|
|
Reference in New Issue
Block a user