Code Monkey home page Code Monkey logo

webgpu-msm's Introduction

webgpu-msm

Quick Start

Ensure you have:

Then run the following:

1) Clone the repository

git clone https://github.com/demox-labs/webgpu-msm && cd webgpu-msm

2) Install dependencies

yarn

3) Development

Run a local server on localhost:4040.

yarn start

Note -- running webgpu functions will only work on browsers compatible with webgpu.

Prize Description

Summary

Multi-Scalar multiplication (MSM) operations are critical for building proofs and zk computations. They are also some of the slowest operations in these processes, especially running on consumer-grade hardware. This z-prize seeks to optimize for consumer-grade devices running msms in a browser environment.

Optimization Objective

Achieve the fastest MSM run in a browser over a range of input vector lengths and a range of consumer hardware.

Constraints

The implementation must provide the following interface in JavaScript: compute_msm = async (baseAffinePoints: BigIntPoint[] | U32ArrayPoint[] | Buffer, scalars: bigint[] | Uint32Array[]| Buffer): Promise<{x: bigint, y: bigint}>

  1. The function name is compute_msm

  2. There are two input vectors: baseAffinePoints for a vector of elliptic curve affine points and scalars for a vector of finite field elements. There are three options for how the inputs are ingested, using bigints or Uint32Arrays or Buffers:

    a.

    BigIntPoint: {
      x: bigint,
      y: bigint,
      t: bigint,
      z: bigint
    }
    scalar: bigint
    

    b.

    BigEndian
    U32ArrayPoint: {
      x: Uint32Array,
      y: Uint32Array,
      t: Uint32Array,
      z: Uint32Array
    }
    scalar: Uint32Array
    

    c.

    LittleEndian
    points: Buffer,
    scalars: Buffer
    

    Note -- the inputs are affine points, so the z property will be 1n or [0, 0, 0, 0, 0, 0, 0, 1]. The t property is the field multiplication of x by y.

  3. The output is a single elliptic curve affine point: { x: bigint, y: bigint }.

  • The submission should produce correct outputs on input vectors with length up to 2^20. The evaluation will be using input randomly sampled from size 2^16 ~ 2^20.

  • The submissions will be evaluated in a browser context on a variety of consumer-grade hardware.

  • All submissions must include documentation (in English) sufficient to understand the approach being taken.

  1. Do I have to provide support for all three of the input types?
  • No. You only need to take one of the three types as inputs to the msm_compute function, and you may pass in whichever input type you'd like to accept in the AllBenchmarks.tsx file. You may cast your inputs to whatever type you'd like to accept, and casting will not affect the runtime performance. For example:
export const compute_msm = async (
  baseAffinePoints: BigIntPoint[] | U32ArrayPoint[] | Buffer,
  scalars: bigint[] | Uint32Array[] | Buffer
  ): Promise<{x: bigint, y: bigint}> => {
  const pointsAsBigIntPoint = baseAffinePoints as BigIntPoint[];
  const pointsAsUint32ArrayBigEndian = baseAffinePoints as Uint32Array[];
  const pointsAsBufferLittleEndian = baseAffinePoints as Buffer;
  .
  .
  .
  }

For passing in whichever input type you'd like to accept, modify the Benchmark for your msm in the AllBenchmarks.tsx file.

<Benchmark
  name={'Your bigInt MSM'}
  disabled={disabledBenchmark}
  baseAffinePoints={baseAffineBigIntPoints}
  scalars={bigIntScalars}
  expectedResult={expectedResult}
  msmFunc={compute_msm}
  postResult={postResult}
  bold={true}
/>
<Benchmark
  name={'Your buffer MSM'}
  disabled={disabledBenchmark}
  baseAffinePoints={bufferPoints}
  scalars={bufferScalars}
  expectedResult={expectedResult}
  msmFunc={compute_msm}
  postResult={postResult}
  bold={true}
/>
<Benchmark
  name={'Your u32 MSM'}
  disabled={disabledBenchmark}
  baseAffinePoints={u32Points}
  scalars={u32Scalars}
  expectedResult={expectedResult}
  msmFunc={compute_msm}
  postResult={postResult}
  bold={true}
/>

Judging

Submissions will be analyzed for both correctness and performance. All submission code must be open-sourced at the time of submission. Code and documentation must be dual-licensed under both the MIT and Apache-2.0 licenses.

Correctness

We have provided test case data and answers to sanity-check your msm function against, as well as multiple msm implementations to check against.

The final correctness of the submission will be tested using randomly sampled test inputs/outputs that are not disclosed to the competitors during the competition in addition to the test input/output distributed to the competitor. All these test cases will be generated using Aleo's reference wasm implementation. Submissions failed in any test cases will be judged as incorrect and lose the opportunity to win the prize.

Performance

To evaluate the performance of each submission, the prize committee will sample a large number of input vectors at random, in terms of both the input vector length and the point values. The input vector length will be sampled between 2^16 and 2^20. Then, we will compute the multi-scalar multiplication using the submission. The submitted score will be the relative speedup from the provided best webgpu msm implementation, averaged across many trials.

In addition, all submissions will be manually reviewed by the prize committee.

Hardware & Benchmarks

The baseline will be the Demox Labs WebGPU MSM implementation over BLS12-377 G1. This baseline is originally implemented in typescript and wgsl. Submissions must beat this baseline by at least 10% in order to be eligible for the prize.

Prize Allocation

Prizes will be given out in good faith and in the sole discretion of the prize committee.

Timeline

  • August 1 - 2023 Registration Opens
  • September 15 - Competition Begins
  • February - Submission Deadline
  • March - Winners Announced

Submission Instruction

Please include your implementation under the submission folder. The compute_msm function in submission/submission.ts will be run when benchmarking your submission.

FAQs

What is the focus of ZPrize Track 2: Beat the Best (Web)?

The goal of this track is to enhance the efficiency of client-side proving for the Aleo proof system using various web technologies.

Can I utilize WebGPU and multithreading (web workers) for optimization?

Yes, you are allowed to use WebGPU and multithreading (web workers) to enhance performance.

What elliptic curve will be used in this competition?

The competition will use BLS12-377 as the elliptic curve.

Is there a preferred web technology for this track? Are there any restrictions on the technologies we can use?

The competition encourages the use of widely supported web technologies. There are no strict restrictions on the technologies you can use, but the focus is on ensuring client proving speed on different devices.

Will participants have access to evaluation platforms for benchmarking? Can all WASM features be utilized?

Access to evaluation platforms will not be provided. The benchmarking aims to test solutions on various commodity devices and seeks general browser environment improvements. Most WASM features that work within Chrome v115 are allowed, including multithreading through web workers.

Is WebGPU included in this prize category?

Yes! WebGPU may be used for any amount of the submission implementation.

Which WebAssembly (Wasm) features are supported for this track?

You can use all of the WASM features that work within chrome v115. Multithreading is allowed for instance but browsers don't have threads so you have to use web workers to accomplish this using something like: https://github.com/GoogleChromeLabs/wasm-bindgen-rayon.

Further Questions

If there are any questions about this prize:

Trouble Shooting

Common issues:

  • If you are unable to run the webgpu benchmarks, ensure you are using a webgpu-compatible browser.
  • If you are not able to load the test case data, be sure you have installed git LFS. You can either reclone the repo after installing git LFS or run git lfs fetch && git lfs pull.
  • If you run into general npm package errors, make sure you have nodejs v16 or later installed.
  • If you are using webgpu functions and getting all 0s as output, you may have hit an out of memory error in the gpu. Reduce your input size or consider breaking your computions into smaller steps.
  • If you are running on a windows machine with multiple graphics cards (ie. integrated chip + dedicated card) you can force Chrome to use one of them via windows settings. Go to Settings -> Graphics Settings -> Custom options for apps -> Select / add Chrome -> Options -> Select graphics preference

Reference

[1] Scalar-multiplication algorithms. https://cryptojedi.org/peter/data/eccss-20130911b.pdf

[2] compute_msm functions in Typescript/WebGPU/WASM https://github.com/demox-labs/webgpu-msm/blob/main/src/reference/reference.tsj

[3] wgsl reference https://www.w3.org/TR/WGSL/

[4] aleo wasm reference https://github.com/demox-labs/aleo/blob/gpu-expose/wasm/src/account/address.rs#L167

webgpu-msm's People

Contributors

evanmarshall avatar fulltimemike avatar johndonavon avatar mitschabaude avatar

Stargazers

 avatar

Watchers

 avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.