Commit e5fce3ca authored by cfhammill's avatar cfhammill
Browse files

python3Packages.openai-triton: fix build module

parent 92e8eb20

pkgs/development/python-modules/openai-triton/default.nix

+1 −2
Original line number Diff line number Diff line
@@ -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