utils.h File Reference

#include <metal_stdlib>

Go to the source code of this file.

Macros
#define	STEEL_CONST   static constant constexpr const
#define	STEEL_PRAGMA_UNROLL   _Pragma("clang loop unroll(full)")

Functions
METAL_FUNC ulong2	elem_to_loc_broadcast (uint elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, int ndim)
METAL_FUNC ulong3	elem_to_loc_broadcast (uint elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, int ndim)

Macro Definition Documentation

STEEL_CONST

#define STEEL_CONST   static constant constexpr const

STEEL_PRAGMA_UNROLL

#define STEEL_PRAGMA_UNROLL   _Pragma("clang loop unroll(full)")

Function Documentation

elem_to_loc_broadcast() [1/2]

METAL_FUNC ulong3 elem_to_loc_broadcast	(	uint	elem,
		constant const int *	shape,
		constant const size_t *	a_strides,
		constant const size_t *	b_strides,
		constant const size_t *	c_strides,
		int	ndim )

elem_to_loc_broadcast() [2/2]

METAL_FUNC ulong2 elem_to_loc_broadcast	(	uint	elem,
		constant const int *	shape,
		constant const size_t *	a_strides,
		constant const size_t *	b_strides,
		int	ndim )