From 0a44bbbb3dcffe68ef71c5d73d5b455970cc5b1e Mon Sep 17 00:00:00 2001
From: Sable Seyler <sable@seyleri.us>
Date: Sat, 5 Jun 2021 19:11:02 -0700
Subject: [PATCH] terra: add toggle argument for CUDA support

Terra can be built with CUDA support, and this commit adds a toggle to
the arguments, `enableCUDA`, for enabling it.
---
 pkgs/development/compilers/terra/default.nix | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pkgs/development/compilers/terra/default.nix b/pkgs/development/compilers/terra/default.nix
index e18f92a4fefa..78c09591df28 100644
--- a/pkgs/development/compilers/terra/default.nix
+++ b/pkgs/development/compilers/terra/default.nix
@@ -1,5 +1,5 @@
 { lib, stdenv, fetchurl, fetchFromGitHub, llvmPackages, ncurses, cmake, libxml2
-, symlinkJoin, breakpointHook }:
+, symlinkJoin, breakpointHook, cudaPackages, enableCUDA ? false }:
 
 let
   luajitRev = "9143e86498436892cb4316550be4d45b68a61224";
@@ -19,6 +19,7 @@ let
       clang-unwrapped.lib
     ];
   };
+  cuda = cudaPackages.cudatoolkit_11;
 in stdenv.mkDerivation rec {
   pname = "terra";
   version = "1.0.0-beta3";
@@ -31,13 +32,13 @@ in stdenv.mkDerivation rec {
   };
 
   nativeBuildInputs = [ cmake ];
-  buildInputs = [ llvmMerged ncurses libxml2 ];
+  buildInputs = [ llvmMerged ncurses libxml2 ] ++ lib.optional enableCUDA cuda;
 
   cmakeFlags = [
     "-DHAS_TERRA_VERSION=0"
     "-DTERRA_VERSION=${src.rev}"
     "-DTERRA_LUA=luajit"
-  ];
+  ] ++ lib.optional enableCUDA "-DTERRA_ENABLE_CUDA=ON";
 
   doCheck = true;
   enableParallelBuilding = true;