gpu::Bindings< N > Struct Template Reference

Represents an ordered collection of WGPUBuffers (wrapped as tensors, non-overlapping views, or arrays) for the purpose of binding them to a kernel operation to make them accessible to the GPU kernel. More...

#include <gpu.h>

Public Member Functions
	Bindings (const std::initializer_list< Tensor > &init)
	Bindings (const std::initializer_list< TensorView > &init)
	Bindings (const std::initializer_list< Array > &init)
Tensor &	operator[] (std::size_t index)
const Tensor &	operator[] (std::size_t index) const

Public Attributes
std::array< Tensor, N >	data
std::array< size_t, N >	viewOffsets
std::array< size_t, N >	viewSpans

Detailed Description

template<std::size_t N>
struct gpu::Bindings< N >

Represents an ordered collection of WGPUBuffers (wrapped as tensors, non-overlapping views, or arrays) for the purpose of binding them to a kernel operation to make them accessible to the GPU kernel.

The ordering of the bindings should match the binding indices in the WGSL code.

Definition at line 124 of file gpu.h.

Constructor & Destructor Documentation

Bindings() [1/3]

template<std::size_t N>

gpu::Bindings< N >::Bindings ( const std::initializer_list< Tensor > & init )

inline

Definition at line 128 of file gpu.h.

                                                    {
    std::copy(begin(init), end(init), begin(data));
    std::fill(begin(viewOffsets), end(viewOffsets), 0);
    for (size_t i = 0; i < N; ++i) {
      viewSpans[i] = data[i].data.size;
    }
  }

Bindings() [2/3]

template<std::size_t N>

gpu::Bindings< N >::Bindings ( const std::initializer_list< TensorView > & init )

inline

Definition at line 136 of file gpu.h.

                                                        {
    size_t i = 0;
    for (const auto &tv : init) {
      data[i] = tv.data;
      viewOffsets[i] = tv.offset;
      viewSpans[i] = tv.span;
      ++i;
    }
  }

Bindings() [3/3]

template<std::size_t N>

gpu::Bindings< N >::Bindings ( const std::initializer_list< Array > & init )

inline

Definition at line 146 of file gpu.h.

                                                   {
    std::copy(begin(init), end(init), begin(data));
    std::fill(begin(viewOffsets), end(viewOffsets), 0);
    for (size_t i = 0; i < N; ++i) {
      viewSpans[i] = data[i].size;
    }
  }

Member Function Documentation

operator[]() [1/2]

template<std::size_t N>

Tensor & gpu::Bindings< N >::operator[] ( std::size_t index )

inline

Definition at line 154 of file gpu.h.

{ return data[index]; }

operator[]() [2/2]

template<std::size_t N>

const Tensor & gpu::Bindings< N >::operator[] ( std::size_t index ) const

inline

Definition at line 155 of file gpu.h.

{ return data[index]; }

Member Data Documentation

data

template<std::size_t N>

std::array<Tensor, N> gpu::Bindings< N >::data

Definition at line 125 of file gpu.h.

viewOffsets

template<std::size_t N>

std::array<size_t, N> gpu::Bindings< N >::viewOffsets

Definition at line 126 of file gpu.h.

viewSpans

template<std::size_t N>

std::array<size_t, N> gpu::Bindings< N >::viewSpans

Definition at line 127 of file gpu.h.

---

The documentation for this struct was generated from the following file:

• gpu.h