Returns a pointer to the managed object or nullptr if no object is owned.

Add position encodings to :obj:`inputs`. Args: inputs: The inputs to encode. position: The single position to encode, to u

* \brief Forward a model and asynchronously check whether to stop. * * The device having the last rank of a pipeline parallel group checks a

< Efficiently disables all accesses guarded by mask

Size of the packed Params structure

Writes the entire listing to a file

Loads a fragment from memory at the location pointed to by the iterator.

Transform the mma operands to the required types

Returns a reference to the element at a given Coord

Establishes an ordering w.r.t previously issued cp.async instructions. Does not block.

Clears the predicate set efficiently

Converts a ExecutionMode enumerant to a string

Obtains the threadblock offset (in units of threadblock-scoped tiles)

Returns whether access is valid or not

< Layout function

Used for initialization

Instantiate a PreTrainedBertModel from a pre-trained model file. Download and cache the pre-trained model file if needed.

Stores a fragment to memory

Unpacks an element from memory

< Layout function

< Layout function

Loads a fragment from memory

Maximum finite value

Returns a const iterator

Gets pointer to host data

Start the timer.

Returns the shape of the problem in units of logical tiles Gemm* problem size: gemm(M, N, K)

Set a bit within the predicate vector.

Accesses the x element of the imaginary part of the quaternion

Accesses the y element of the imaginary part of the quaternion

Accesses the z element of the imaginary part of the quaternion

Bitcasts to CUDA's half type

Returns the complex conjugate

Stop the timer.

Store a fragment to memory

Gets the type of the iterator

Clear the cache

Advances an iterator along logical dimensions of matrix in units of whole threadblock tiles

Returns true if element is in a defined state

Ensure that numerator is divisible by the denominator and return the division value.

Filtering operations based on various criteria

Converts to float

Obtains a Coord<2> from GemmCoord

Returns the layout object