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

Author: monyarm

flake.nix

@@ -31,7 +31,12 @@
   };
 
   outputs =
-    inputs@{ flake-parts, nixos-modules, ... }:
+    inputs@{
+      flake-parts,
+      nixos-modules,
+      nixpkgs,
+      ...
+    }:
     flake-parts.lib.mkFlake { inherit inputs; } {
       systems = [
         "x86_64-linux"
@@ -48,9 +53,14 @@
           final,
           self',
           config,
+          system,
           ...
         }:
         {
+          _module.args.pkgs = import nixpkgs {
+            inherit system;
+            config.allowUnfree = true;
+          };
           devShells.default = import ./shells/all.nix {
             pkgs = final;
             inherit self';

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,56 @@
+{
+  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;
+  };
+}