TilePipeline

class nvmath.device.TilePipeline(device_pipeline: DevicePipeline)

TilePipeline allows users to execute an pipelined matrix multiplication with partial tile results accumulated into an acuumulator.

Refer to the cuBLASDx documentation for more details on how to use this class: https://docs.nvidia.com/cuda/cublasdx/using_pipelines.html

Methods

__init__(

device_pipeline: DevicePipeline,

)

execute(accumulator)

Attributes

storage_alignment

storage_bytes