openvm-circuit-primitives 2.0.2

Library of plonky3 primitives for general purpose use in other ZK circuits.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#pragma once

#include <cstdint>
#include <cassert>

template <typename T>
struct DeviceBufferConstView {
    T const* ptr;
    size_t size;

    __device__ __host__ __forceinline__ T const* begin() const {
        return ptr;
    }

    __device__ __host__ __forceinline__ T const* end() const {
        return ptr + len();
    }

    __device__ __host__ __forceinline__ T const& operator [](size_t idx) const {
        assert(idx < len());
        return ptr[idx];
    }

    __device__ __host__ __forceinline__ size_t len() const {
        return size / sizeof(T);
    }
};

struct DeviceRawBufferConstView {
    uintptr_t ptr;
    size_t size;

    template <typename T> __device__ __host__ DeviceBufferConstView<T> as_typed() const {
        assert(size % sizeof(T) == 0);
        return {
            reinterpret_cast<T const*>(ptr),
            size
        };
    }
};