Mojo struct

BlackwellProfileWarp

struct BlackwellProfileWarp[load_warps: UInt32, mma_warps: UInt32, scheduler_warps: UInt32, epilogue_warps: UInt32, max_entries_per_warp: UInt32, //, WorkspaceManager: BlackwellWarpProfilingWorkspaceManager[load_warps, mma_warps, scheduler_warps, epilogue_warps, max_entries_per_warp], warp_role: UInt32 = UInt32(0)]

This struct calculates execution time for a warp/s, and writes a single entry to the workspace.

Fields

timeline (Tuple[UInt64, UInt64]):
workspace (Span[UInt64, MutAnyOrigin]):
entry_idx (UInt32):

Implemented traits

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDestructible, Movable

`comptime` members

`enable_profiling`

comptime enable_profiling = (max_entries_per_warp > UInt32(0))

Methods

`init`

__init__(out self, workspace: Span[UInt64, MutAnyOrigin], entry_idx: UInt32)

`enter`

__enter__(mut self)

`exit`

__exit__(mut self)

Fields​

Implemented traits​

comptime members​

enable_profiling​

Methods​

__init__​

__enter__​

__exit__​