/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/library/utility/device_tensor_generator.hpp File Reference

• develop
• include
• ck
• library
• utility

Classes | Functions

device_tensor_generator.hpp File Reference

#include <hip/hip_runtime.h>
#include "ck/ck.hpp"
#include "ck/utility/common_header.hpp"
#include "ck/library/utility/device_tensor_generator.hpp"
#include "ck/utility/data_type.hpp"

Go to the source code of this file.

Classes
struct	ran_state_u32

Functions
__device__ uint32_t	ran_gen_round_u32 (ran_state_u32 &state)
__device__ ran_state_u32	ran_init (uint32_t seed=0)
template<typename T >
__global__ void	fill_tensor_uniform_rand_int_values (T *p, int min_value, int max_value, uint64_t buffer_element_size)
template<typename T >
__global__ void	fill_tensor_uniform_rand_fp_values (T *p, float min_value, float max_value, uint64_t buffer_element_size)
template<typename T >
__global__ void	fill_tensor_norm_rand_fp_values (T *p, float sigma, float mean, uint64_t buffer_element_size)

Function Documentation

◆ fill_tensor_norm_rand_fp_values()

template<typename T >

__global__ void fill_tensor_norm_rand_fp_values	(	T *	p,
		float	sigma,
		float	mean,
		uint64_t	buffer_element_size
	)

◆ fill_tensor_uniform_rand_fp_values()

template<typename T >

__global__ void fill_tensor_uniform_rand_fp_values	(	T *	p,
		float	min_value,
		float	max_value,
		uint64_t	buffer_element_size
	)

◆ fill_tensor_uniform_rand_int_values()

template<typename T >

__global__ void fill_tensor_uniform_rand_int_values	(	T *	p,
		int	min_value,
		int	max_value,
		uint64_t	buffer_element_size
	)

◆ ran_gen_round_u32()

__device__ uint32_t ran_gen_round_u32

(

ran_state_u32 &

state

)

◆ ran_init()

__device__ ran_state_u32 ran_init

(

uint32_t

seed = 0

)