elementwise_scale.hpp File Reference

Compiled CUDA entry points for elementwise scaling. More...

#include <uni20/core/types.hpp>
#include <uni20/linalg/backends/cuda/elementwise_plan.hpp>
#include <cuda_runtime_api.h>
#include <concepts>
#include <cstddef>
#include <type_traits>

Include dependency graph for elementwise_scale.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	uni20
namespace	uni20::linalg

Macros
#define	UNI20_DECLARE_ELEMENTWISE_SCALE(Scalar, Factor)

Detailed Description

Compiled CUDA entry points for elementwise scaling.

Macro Definition Documentation

UNI20_DECLARE_ELEMENTWISE_SCALE

#define UNI20_DECLARE_ELEMENTWISE_SCALE	(		Scalar,
			Factor
	)

Value:

  void enqueue_elementwise_scale(Scalar* output, Scalar const* input, Factor factor,                                   \
                                 ElementwiseScalePlan32 const& plan, cudaStream_t stream, int device);                 \
  void enqueue_elementwise_scale(Scalar* output, Scalar const* input, Factor factor,                                   \
                                 ElementwiseScalePlan64 const& plan, cudaStream_t stream, int device)