TensorView结构体 参考

#include <slang-cuda-prelude.h>

Public 成员函数
template<typename T >
__device__ T *	data_ptr ()
template<typename T >
__device__ T *	data_ptr_at (uint32_t index)
template<typename T >
__device__ T *	data_ptr_at (uint2 index)
template<typename T >
__device__ T *	data_ptr_at (uint3 index)
template<typename T >
__device__ T *	data_ptr_at (uint4 index)
template<typename T , unsigned int N>
__device__ T *	data_ptr_at (uint index[N])
template<typename T >
__device__ T &	load (uint32_t x)
template<typename T >
__device__ T &	load (uint32_t x, uint32_t y)
template<typename T >
__device__ T &	load (uint2 index)
template<typename T >
__device__ T &	load (uint32_t x, uint32_t y, uint32_t z)
template<typename T >
__device__ T &	load (uint3 index)
template<typename T >
__device__ T &	load (uint32_t x, uint32_t y, uint32_t z, uint32_t w)
template<typename T >
__device__ T &	load (uint4 index)
template<typename T >
__device__ T &	load (uint32_t i0, uint32_t i1, uint32_t i2, uint32_t i3, uint32_t i4)
template<typename T , unsigned int N>
__device__ T &	load (uint index[N])
template<typename T >
__device__ void	store (uint32_t x, T val)
template<typename T >
__device__ void	store (uint32_t x, uint32_t y, T val)
template<typename T >
__device__ void	store (uint2 index, T val)
template<typename T >
__device__ void	store (uint32_t x, uint32_t y, uint32_t z, T val)
template<typename T >
__device__ void	store (uint3 index, T val)
template<typename T >
__device__ void	store (uint32_t x, uint32_t y, uint32_t z, uint32_t w, T val)
template<typename T >
__device__ void	store (uint4 index, T val)
template<typename T >
__device__ void	store (uint32_t i0, uint32_t i1, uint32_t i2, uint32_t i3, uint32_t i4, T val)
template<typename T , unsigned int N>
__device__ void	store (uint index[N], T val)

Public 属性
uint8_t *	data
uint32_t	strides [kSlangTorchTensorMaxDim]
uint32_t	sizes [kSlangTorchTensorMaxDim]
uint32_t	dimensionCount

成员函数说明

data_ptr()

template<typename T >

__device__ T * TensorView::data_ptr ( )

inline

data_ptr_at() [1/5]

template<typename T , unsigned int N>

__device__ T * TensorView::data_ptr_at ( uint index[N] )

inline

data_ptr_at() [2/5]

template<typename T >

__device__ T * TensorView::data_ptr_at ( uint2 index )

inline

data_ptr_at() [3/5]

template<typename T >

__device__ T * TensorView::data_ptr_at ( uint3 index )

inline

data_ptr_at() [4/5]

template<typename T >

__device__ T * TensorView::data_ptr_at ( uint32_t index )

inline

data_ptr_at() [5/5]

template<typename T >

__device__ T * TensorView::data_ptr_at ( uint4 index )

inline

load() [1/9]

template<typename T , unsigned int N>

__device__ T & TensorView::load ( uint index[N] )

inline

load() [2/9]

template<typename T >

__device__ T & TensorView::load ( uint2 index )

inline

load() [3/9]

template<typename T >

__device__ T & TensorView::load ( uint3 index )

inline

load() [4/9]

template<typename T >

__device__ T & TensorView::load ( uint32_t i0, uint32_t i1, uint32_t i2, uint32_t i3, uint32_t i4 )

inline

load() [5/9]

template<typename T >

__device__ T & TensorView::load ( uint32_t x )

inline

load() [6/9]

template<typename T >

__device__ T & TensorView::load ( uint32_t x, uint32_t y )

inline

load() [7/9]

template<typename T >

__device__ T & TensorView::load ( uint32_t x, uint32_t y, uint32_t z )

inline

load() [8/9]

template<typename T >

__device__ T & TensorView::load ( uint32_t x, uint32_t y, uint32_t z, uint32_t w )

inline

load() [9/9]

template<typename T >

__device__ T & TensorView::load ( uint4 index )

inline

store() [1/9]

template<typename T , unsigned int N>

__device__ void TensorView::store ( uint index[N], T val )

inline

store() [2/9]

template<typename T >

__device__ void TensorView::store ( uint2 index, T val )

inline

store() [3/9]

template<typename T >

__device__ void TensorView::store ( uint3 index, T val )

inline

store() [4/9]

template<typename T >

__device__ void TensorView::store ( uint32_t i0, uint32_t i1, uint32_t i2, uint32_t i3, uint32_t i4, T val )

inline

store() [5/9]

template<typename T >

__device__ void TensorView::store ( uint32_t x, T val )

inline

store() [6/9]

template<typename T >

__device__ void TensorView::store ( uint32_t x, uint32_t y, T val )

inline

store() [7/9]

template<typename T >

__device__ void TensorView::store ( uint32_t x, uint32_t y, uint32_t z, T val )

inline

store() [8/9]

template<typename T >

__device__ void TensorView::store ( uint32_t x, uint32_t y, uint32_t z, uint32_t w, T val )

inline

store() [9/9]

template<typename T >

__device__ void TensorView::store ( uint4 index, T val )

inline

类成员变量说明

data

uint8_t * TensorView::data

dimensionCount

uint32_t TensorView::dimensionCount

sizes

uint32_t TensorView::sizes

strides

uint32_t TensorView::strides

---

该结构体的文档由以下文件生成:

• third/slang/include/slang-cuda-prelude.h
• third/slang/include/slang-torch-prelude.h