python3Packages.openai-triton: fix build module

This commit is contained in:
cfhammill
2024-07-09 09:28:15 -04:00
parent 92e8eb20a4
commit e5fce3ca48
@@ -91,8 +91,7 @@ buildPythonPackage rec {
--replace "include (\''${CMAKE_CURRENT_SOURCE_DIR}/googletest.cmake)" ""\
--replace "include(GoogleTest)" "find_package(GTest REQUIRED)"
cat << \EOF > python/triton/common/build.py
cat << \EOF >> python/triton/common/build.py
def libcuda_dirs():
return [ "${addDriverRunpath.driverLink}/lib" ]
EOF