feat(rapidsnark-gpu): Add rapidsnark-gpu package

Author: monyarm

packages/all-packages.nix

@@ -68,6 +68,9 @@
       rapidsnark = callPackage ./rapidsnark/default.nix {
         inherit ffiasm zqfield-bn254;
       };
+      rapidsnark-gpu = callPackage ./rapidsnark-gpu/default.nix {
+        inherit ffiasm zqfield-bn254;
+      };
 
       # Elrond / MultiversX
       # copied from https://github.com/NixOS/nixpkgs/blob/8df7949791250b580220eb266e72e77211bedad9/pkgs/development/python-modules/cryptography/default.nix
@@ -223,6 +226,7 @@
             ffiasm
             ffiasm-src
             rapidsnark
+            rapidsnark-gpu
             ;
 
           inherit cardano graphql;

packages/rapidsnark-gpu/default.nix

@@ -0,0 +1,50 @@
+{
+  lib,
+  cudaPackages,
+  stdenv,
+  ffiasm,
+  zqfield-bn254,
+  nlohmann_json,
+  gmp,
+  libsodium,
+  cmake,
+  fetchFromGitHub,
+  pkg-config,
+}: let
+  ffiasm-c = "${ffiasm}/lib/node_modules/ffiasm/c";
+in
+  stdenv.mkDerivation rec {
+    pname = "rapidsnark-gpu";
+    version = "2023-04-08";
+
+    src = fetchFromGitHub {
+      owner = "Orbiter-Finance";
+      repo = "rapidsnark";
+      rev = "77016322808ac58a3acd25a6235510b55172f967";
+      hash = "sha256-8vy+iXkSINFregve+rej1rXyXdWxm0n1wvYfoy/0idk=";
+    };
+
+    nativeBuildInputs = [pkg-config];
+    buildInputs = [nlohmann_json gmp libsodium cudaPackages.cudatoolkit] ++ ffiasm.passthru.openmp;
+
+    buildPhase = ''
+      mkdir -p $out/bin
+      c++ \
+        -I{${ffiasm-c},${zqfield-bn254}/lib} \
+        ./src/{main_prover,binfile_utils,zkey_utils,wtns_utils,logger}.cpp \
+        ${ffiasm-c}/{alt_bn128,misc,naf,splitparstr}.cpp \
+        ${zqfield-bn254}/lib/{fq,fr}.{cpp,o} \
+        $(pkg-config --cflags --libs libsodium gmp nlohmann_json) \
+        -std=c++17 -pthread -O3 -fopenmp \
+        -o $out/bin/prover
+    '';
+
+    installPhase = ''
+      # Already done in buildPhase
+    '';
+
+    meta = {
+      homepage = "https://github.com/iden3/rapidsnark";
+      platforms = with lib.platforms; linux ++ darwin;
+    };
+  }