numerics/api/src_2core_2backends_2gpu_2impl_8hpp_source.html

/// @file core/backends/gpu/impl.hpp

/// @brief Private declarations for the GPU (CUDA) backend.

/// Only included by src/core/vector.cpp and src/core/matrix.cpp.

#pragma once

#include "core/vector.hpp"

#include "core/matrix.hpp"


namespace num::backends::gpu {


void scale(Vector& v, real alpha);

void axpy(real alpha, const Vector& x, Vector& y);

real dot(const Vector& x, const Vector& y);

real norm(const Vector& x);


void matmul(const Matrix& A, const Matrix& B, Matrix& C);

void matvec(const Matrix& A, const Vector& x, Vector& y);


} // namespace num::backends::gpu


num::BasicVector< real >

num::Matrix
Dense row-major matrix with optional GPU storage.
Definition matrix.hpp:12

matrix.hpp
Matrix operations.

num::backends::gpu
Definition impl.hpp:8

num::backends::gpu::axpy
void axpy(real alpha, const Vector &x, Vector &y)
Definition vector.cpp:22

num::backends::gpu::matmul
void matmul(const Matrix &A, const Matrix &B, Matrix &C)
Definition matrix.cpp:13

num::backends::gpu::dot
real dot(const Vector &x, const Vector &y)
Definition vector.cpp:30

num::backends::gpu::matvec
void matvec(const Matrix &A, const Vector &x, Vector &y)
Definition matrix.cpp:22

num::backends::gpu::norm
real norm(const Vector &x)
Definition vector.cpp:38

num::backends::gpu::scale
void scale(Vector &v, real alpha)
Definition vector.cpp:14

num::real
double real
Definition types.hpp:10

num::ipow
constexpr T ipow(T x) noexcept
Compute x^N at compile time via repeated squaring.
Definition integer_pow.hpp:34

vector.hpp
Vector operations.