Blas.cuh File Reference

BLAS API wrapper over cuBLAS. More...

#include "Matrix.cuh"
#include "pbat/gpu/impl/common/Cuda.cuh"
#include <cstdio>
#include <cublas_v2.h>
#include <cuda/api/stream.hpp>
#include <exception>
#include <memory>
#include <type_traits>

Go to the source code of this file.

Classes
class	pbat::gpu::impl::math::Blas

Namespaces
namespace	pbat
	The main namespace of the library.
namespace	pbat::gpu
	GPU related public functionality.
namespace	pbat::gpu::impl
	GPU algorithm implementations.
namespace	pbat::gpu::impl::math
	GPU implementations of math functions.

Macros
#define	PBAT_GPU_IMPL_MATH_BLAS_H
#define	CUBLAS_CHECK(err)

Detailed Description

BLAS API wrapper over cuBLAS.

Author

Quoc-Minh Ton-That (tonth.nosp@m.at.q.nosp@m.uocmi.nosp@m.nh@g.nosp@m.mail..nosp@m.com)

Date

2025-04-24

Copyright

Copyright (c) 2025

Macro Definition Documentation

CUBLAS_CHECK

#define CUBLAS_CHECK

(

err

)

Value:

    {                                                                            \
        cublasStatus_t err_ = (err);                                             \
        if (err_ != cublasStatus_t::CUBLAS_STATUS_SUCCESS)                       \
        {                                                                        \
            std::printf("cublas error %d at %s:%d\n", err_, __FILE__, __LINE__); \
            throw std::runtime_error("cublas error");                            \
        }                                                                        \
    }