CUDA Scalar Mul

.github/workflows/ci.yml

@@ -22,7 +22,7 @@ jobs:
         toolchain: stable
         override: true
         components: rustfmt
-
+        default: true
     - name: cargo fmt --check
       uses: actions-rs/cargo@v1
       with:
@@ -35,6 +35,7 @@ jobs:
     env:
       RUSTFLAGS: -Dwarnings
     strategy:
+      max-parallel: 6
       matrix:
         rust:
           - stable
@@ -50,14 +51,38 @@ jobs:
           toolchain: ${{ matrix.rust }}
           override: true
 
-      - uses: actions/cache@v2
-        with:
-          path: |
-            ~/.cargo/registry
-            ~/.cargo/git
-            target
+      - name: Install CUDA toolchains
+        run: |
+          wget -q https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
+          sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
+          wget -q https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda-repo-ubuntu1804-11-1-local_11.1.1-455.32.00-1_amd64.deb
+          sudo dpkg -i cuda-repo-ubuntu1804-11-1-local_11.1.1-455.32.00-1_amd64.deb
+          sudo apt-key add /var/cuda-repo-ubuntu1804-11-1-local/7fa2af80.pub
+          sudo apt-get update
+          sudo apt-get -y install cuda
+          rm cuda-repo-ubuntu*
+          curl -sSL https://github.com/jon-chuang/accel/raw/master/setup_nvptx_toolchain.sh | bash
+
+      - uses: actions/cache@v2	
+        with:	
+          path: |	
+            ~/.cargo/registry	
+            ~/.cargo/git	
+            target	
           key: ${{ runner.os }}-cargo-${{ hashFiles('**/Cargo.lock') }}
 
+      - name: Test algebra with CUDA
+        run: |
+          cd algebra
+          cargo test --features "all_curves cuda cuda_test"
+          cd ..
+
+      - name: Test algebra
+        run: |
+          cd algebra
+          cargo test --features full
+          cd ..
+
       - name: Check examples
         uses: actions-rs/cargo@v1
         with:
@@ -68,7 +93,7 @@ jobs:
         uses: actions-rs/cargo@v1
         with:
             command: check
-            args: --examples --all-features --all
+            args: --all-features --examples --all
         if: matrix.rust == 'stable'
 
       - name: Check benchmarks on nightly
@@ -88,12 +113,6 @@ jobs:
                    --exclude ff-fft-benches \
                    -- --skip dpc --skip integration_test"
 
-      - name: Test algebra
-        run: |
-          cd algebra
-          cargo test --features full
-          cd ..
-
       - name: Test algebra with assembly
         run: |
           cd algebra

Cargo.toml

@@ -15,7 +15,7 @@ members = [
     "r1cs-core",
     "r1cs-std",
     "algebra-core/algebra-core-derive",
-    "scripts/glv_lattice_basis"
+    "scripts/glv_lattice_basis",
 ]
 
 [profile.release]

README.md

@@ -87,6 +87,13 @@ To bench `algebra-benches` with greater accuracy, especially for functions with
 cargo +nightly bench --features "n_fold bls12_381"
 ```
 
+CUDA support is available for a limited set of functions. To allow compilation for CUDA on Linux, first run the script
+```
+curl -sSL https://github.com/jon-chuang/accel/raw/master/setup_nvptx_toolchain.sh | bash
+```
+or run the equivalent commands for your OS. Then, pass the `cuda` feature to rustc or cargo when compiling, and import the relevant traits (e.g. GPUScalarMulSlice) wherever the functions are called.
+
+When the `cuda` feature is not activated, Zexe will still compile. However, when either the `cuda` feature is not activated during compilation or CUDA is not detected on your system at runtime, Zexe will default to a CPU-only implementation of the same functionality.
 
 ## License
 

algebra-benches/Cargo.toml

@@ -31,9 +31,10 @@ rand_xorshift = { version = "0.2" }
 paste = "1.0"
 
 [features]
+bw6_asm = [ "algebra/bw6_asm"]
 asm = [ "algebra/asm"]
 prefetch = [ "algebra/prefetch"]
-bw6_asm = [ "algebra/bw6_asm"]
+cuda = [ "algebra/cuda" ]
 n_fold = []
 mnt4_298 = [ "algebra/mnt4_298"]
 mnt6_298 = [ "algebra/mnt6_298"]

algebra-core/Cargo.toml

@@ -27,32 +27,40 @@ algebra-core-derive = { path = "algebra-core-derive", optional = true }
 derivative = { version = "2", features = ["use_core"] }
 num-traits = { version = "0.2", default-features = false }
 rand = { version = "0.7", default-features = false }
-rayon = { version = "1", optional = true }
+rayon = { version = "1.3.0", optional = true }
 unroll = { version = "=0.1.4" }
 itertools = { version = "0.9.0", default-features = false }
 either = { version = "1.6.0", default-features = false }
 thread-id = { version = "3.3.0", optional = true }
 backtrace = { version = "0.3", optional = true }
+accel = { git = "https://github.com/jon-chuang/accel", package = "accel", optional = true }
+peekmore = "0.5.6"
+closure = { version = "0.3.0", optional = true }
+lazy_static = { version = "1.4.0", optional = true }
+serde_json = { version = "1.0.58", optional = true }
+dirs = { version = "1.0.5", optional = true }
+log = { version = "0.4.11", optional = true }
 paste = "0.1"
 
 [build-dependencies]
 field-assembly = { path = "./field-assembly", optional = true }
-cc = "1.0"
 rustc_version = "0.2"
+cc = "1.0"
 
 [dev-dependencies]
 rand_xorshift = "0.2"
 
 [features]
+bw6_asm = []
 default = [ "std", "rand/default" ]
 std = []
 parallel = [ "std", "rayon", "rand/default" ]
 derive = [ "algebra-core-derive" ]
 prefetch = [ "std" ]
+cuda = [ "std", "parallel", "accel", "lazy_static", "serde_json", "dirs", "closure", "log" ]
 
 timing = [ "std", "backtrace" ]
 timing_detailed = [ "std", "backtrace" ]
 timing_thread_id = [ "thread-id" ]
 
 llvm_asm = [ "field-assembly" ]
-bw6_asm = []

algebra-core/algebra-core-derive/Cargo.toml

@@ -27,4 +27,4 @@ proc-macro = true
 [dependencies]
 proc-macro2 = "1.0"
 syn = "1.0"
-quote = "1.0"
+quote = "1.0.7"

algebra-core/mince/Cargo.toml

@@ -7,7 +7,7 @@ edition = "2018"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-quote = "1.0"
+quote = "1.0.7"
 syn = {version = "1.0.17", features = ["full"]}
 
 [lib]

algebra-core/src/bytes.rs

@@ -316,7 +316,7 @@ mod test {
     fn test_macro_empty() {
         let array: Vec<u8> = vec![];
         let bytes: Vec<u8> = to_bytes![array].unwrap();
-        assert_eq!(&bytes, &[]);
+        assert_eq!(bytes, Vec::<u8>::new());
         assert_eq!(bytes.len(), 0);
     }
 

algebra-core/src/curves/batch_arith.rs

@@ -25,7 +25,7 @@ pub trait BatchGroupArithmetic
 where
     Self: Sized + Clone + Copy + Zero + Neg<Output = Self>,
 {
-    type BBaseField: Field;
+    type BaseFieldForBatch: Field;
 
     // We use the w-NAF method, achieving point density of approximately 1/(w + 1)
     // and requiring storage of only 2^(w - 1).
@@ -136,7 +136,7 @@ where
     fn batch_double_in_place(
         bases: &mut [Self],
         index: &[u32],
-        scratch_space: Option<&mut Vec<Self::BBaseField>>,
+        scratch_space: Option<&mut Vec<Self::BaseFieldForBatch>>,
     );
 
     /// Mutates bases in place and stores result in the first operand.

algebra-core/src/curves/cuda/accel_dummy.rs

@@ -0,0 +1,9 @@
+#[cfg(not(feature = "std"))]
+use alloc::vec::Vec;
+pub mod error {
+    pub type Result<T> = T;
+}
+
+pub struct Context {}
+
+pub type DeviceMemory<T> = Vec<T>;

algebra-core/src/curves/cuda/mod.rs

@@ -0,0 +1,6 @@
+#[macro_use]
+pub mod scalar_mul;
+pub use scalar_mul::*;
+
+#[cfg(not(feature = "cuda"))]
+pub mod accel_dummy;