mlir.dialects._nvvm_enum_gen¶

Attributes¶

_ods_ir

Classes¶

`BarrierReduction`	NVVM barrier reduction operation
`CTAGroupKind`	NVVM CTA group kind
`CacheEvictionPriority`	NVVM Cache Eviction Priority
`ClusterLaunchControlQueryType`	NVVM ClusterLaunchControlQueryType
`DotAccumulateType`	NVVM DotAccumulateType
`FPRoundingMode`	NVVM FPRoundingMode kind
`GridDepActionKind`	Action kind for grid dependency control
`LdStMatrixEltType`	Element type for ldmatrix and stmatrix
`LoadCacheModifierKind`	NVVM load cache modifier kind
`MMAB1Op`	MMA binary operations
`MMAFrag`	NVVM MMA frag type
`MMAIntOverflow`	MMA overflow options
`MMALayout`	NVVM MMA layout
`MMATypes`	NVVM MMA types
`MatchSyncKind`	NVVM match sync kind
`MemScopeKind`	NVVM Memory Scope kind
`NVVMMemorySpace`	NVVM Memory Space
`PrefetchCacheLevel`	NVVM Prefetch Cache Level
`ProxyKind`	Proxy kind
`ReduxKind`	NVVM redux kind
`SaturationMode`	NVVM SaturationMode kind
`SetMaxRegisterAction`	NVVM set max register action
`SharedSpace`	Shared memory space
`ShflKind`	NVVM shuffle kind
`TMALoadMode`	NVVM TMA Load Mode
`TMAReduxKind`	NVVM TMA redux kind
`TMAStoreMode`	NVVM TMA Store Mode
`Tcgen05CpMulticast`	tcgen05 cp multicast
`Tcgen05CpShape`	tcgen05 cp shapes
`Tcgen05CpSrcFormat`	tcgen05 cp source format
`Tcgen05FenceKind`	NVVM Tcgen05 fence kind
`Tcgen05LdStShape`	allowed 32-bit signless integer cases: 0, 1, 2, 3, 4
`Tcgen05WaitKind`	NVVM Tcgen05 wait kind
`VoteSyncKind`	NVVM vote sync kind
`WGMMAScaleIn`	WGMMA overflow options
`WGMMAScaleOut`	WGMMA input predicate
`WGMMATypes`	NVVM WGMMA types

Functions¶

`_barrierreduction`(x, context)
`_ctagroupkind`(x, context)
`_dotaccumulatetype`(x, context)
`_fproundingmode`(x, context)
`_ldstmatrixelttype`(x, context)
`_loadcachemodifierkind`(x, context)
`_mmab1op`(x, context)
`_mmafrag`(x, context)
`_mmaintoverflow`(x, context)
`_mmalayout`(x, context)
`_mmatypes`(x, context)
`_matchsynckind`(x, context)
`_memscopekind`(x, context)
`_proxykind`(x, context)
`_reduxkind`(x, context)
`_saturationmode`(x, context)
`_setmaxregisteraction`(x, context)
`_sharedspace`(x, context)
`_shflkind`(x, context)
`_tmaloadmode`(x, context)
`_tmareduxkind`(x, context)
`_tmastoremode`(x, context)
`_tcgen05cpmulticast`(x, context)
`_tcgen05cpshape`(x, context)
`_tcgen05cpsrcformat`(x, context)
`_tcgen05fencekind`(x, context)
`_tcgen05ldstshape`(x, context)
`_tcgen05waitkind`(x, context)
`_votesynckind`(x, context)
`_wgmmascalein`(x, context)
`_wgmmascaleout`(x, context)
`_wgmmatypes`(x, context)
`_barrierreductionattr`(x, context)
`_ctagroupkindattr`(x, context)
`_cacheevictionpriorityattr`(x, context)
`_clusterlaunchcontrolquerytypeattr`(x, context)
`_dotaccumulatetypeattr`(x, context)
`_fproundingmodeattr`(x, context)
`_griddepactionattr`(x, context)
`_ldstmatrixelttypeattr`(x, context)
`_loadcachemodifierattr`(x, context)
`_mmab1opattr`(x, context)
`_mmafragattr`(x, context)
`_mmaintoverflowattr`(x, context)
`_mmalayoutattr`(x, context)
`_mmatypesattr`(x, context)
`_matchsynckindattr`(x, context)
`_memscopekindattr`(x, context)
`_nvvmmemoryspaceattr`(x, context)
`_prefetchcachelevelattr`(x, context)
`_proxykindattr`(x, context)
`_reduxkindattr`(x, context)
`_saturationmodeattr`(x, context)
`_setmaxregisteractionattr`(x, context)
`_sharedspaceattr`(x, context)
`_shflkindattr`(x, context)
`_tmaloadmodeattr`(x, context)
`_tmareduxkindattr`(x, context)
`_tmastoremodeattr`(x, context)
`_tcgen05cpmulticastattr`(x, context)
`_tcgen05cpshapeattr`(x, context)
`_tcgen05cpsrcformatattr`(x, context)
`_tcgen05fencekindattr`(x, context)
`_tcgen05ldstshapeattr`(x, context)
`_tcgen05waitkindattr`(x, context)
`_votesynckindattr`(x, context)
`_wgmmascaleinattr`(x, context)
`_wgmmascaleoutattr`(x, context)
`_wgmmatypesattr`(x, context)

Module Contents¶

mlir.dialects._nvvm_enum_gen._ods_ir¶

class mlir.dialects._nvvm_enum_gen.BarrierReduction¶

Bases: enum.IntEnum

NVVM barrier reduction operation

POPC = 0¶

AND = 1¶

OR = 2¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._barrierreduction(x, context)¶

class mlir.dialects._nvvm_enum_gen.CTAGroupKind¶

Bases: enum.IntEnum

NVVM CTA group kind

CTA_1 = 0¶

CTA_2 = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._ctagroupkind(x, context)¶

class mlir.dialects._nvvm_enum_gen.CacheEvictionPriority¶

Bases: enum.IntEnum

NVVM Cache Eviction Priority

EvictNormal = 0¶

EvictFirst = 1¶

EvictLast = 2¶

EvictUnchanged = 3¶

NoAllocate = 4¶

__str__()¶: Return str(self).

class mlir.dialects._nvvm_enum_gen.ClusterLaunchControlQueryType¶

Bases: enum.IntEnum

NVVM ClusterLaunchControlQueryType

IS_CANCELED = 0¶

GET_FIRST_CTA_ID_X = 1¶

GET_FIRST_CTA_ID_Y = 2¶

GET_FIRST_CTA_ID_Z = 3¶

__str__()¶: Return str(self).

class mlir.dialects._nvvm_enum_gen.DotAccumulateType¶

Bases: enum.IntEnum

NVVM DotAccumulateType

SIGNED = 1¶

UNSIGNED = 0¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._dotaccumulatetype(x, context)¶

class mlir.dialects._nvvm_enum_gen.FPRoundingMode¶

Bases: enum.IntEnum

NVVM FPRoundingMode kind

NONE = 0¶

RN = 1¶

RM = 2¶

RP = 3¶

RZ = 4¶

RNA = 5¶

RS = 6¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._fproundingmode(x, context)¶

class mlir.dialects._nvvm_enum_gen.GridDepActionKind¶

Bases: enum.IntEnum

Action kind for grid dependency control

wait = 0¶

launch_dependents = 1¶

__str__()¶: Return str(self).

class mlir.dialects._nvvm_enum_gen.LdStMatrixEltType¶

Bases: enum.IntEnum

Element type for ldmatrix and stmatrix

B16 = 0¶

B8 = 1¶

B8X16_B6X16_P32 = 2¶

B8X16_B4X16_P64 = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._ldstmatrixelttype(x, context)¶

class mlir.dialects._nvvm_enum_gen.LoadCacheModifierKind¶

Bases: enum.IntEnum

NVVM load cache modifier kind

CA = 0¶

CG = 1¶

CS = 2¶

LU = 3¶

CV = 4¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._loadcachemodifierkind(x, context)¶

class mlir.dialects._nvvm_enum_gen.MMAB1Op¶

Bases: enum.IntEnum

MMA binary operations

none = 0¶

xor_popc = 1¶

and_popc = 2¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._mmab1op(x, context)¶

class mlir.dialects._nvvm_enum_gen.MMAFrag¶

Bases: enum.IntEnum

NVVM MMA frag type

a = 0¶

b = 1¶

c = 2¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._mmafrag(x, context)¶

class mlir.dialects._nvvm_enum_gen.MMAIntOverflow¶

Bases: enum.IntEnum

MMA overflow options

satfinite = 1¶

wrapped = 0¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._mmaintoverflow(x, context)¶

class mlir.dialects._nvvm_enum_gen.MMALayout¶

Bases: enum.IntEnum

NVVM MMA layout

row = 0¶

col = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._mmalayout(x, context)¶

class mlir.dialects._nvvm_enum_gen.MMATypes¶

Bases: enum.IntEnum

NVVM MMA types

f16 = 0¶

f32 = 1¶

tf32 = 2¶

bf16 = 9¶

s8 = 4¶

u8 = 3¶

s32 = 5¶

s4 = 8¶

u4 = 7¶

b1 = 6¶

f64 = 10¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._mmatypes(x, context)¶

class mlir.dialects._nvvm_enum_gen.MatchSyncKind¶

Bases: enum.IntEnum

NVVM match sync kind

any = 0¶

all = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._matchsynckind(x, context)¶

class mlir.dialects._nvvm_enum_gen.MemScopeKind¶

Bases: enum.IntEnum

NVVM Memory Scope kind

CTA = 0¶

CLUSTER = 1¶

GPU = 2¶

SYS = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._memscopekind(x, context)¶

class mlir.dialects._nvvm_enum_gen.NVVMMemorySpace¶

Bases: enum.IntEnum

NVVM Memory Space

Generic = 0¶

Global = 1¶

Shared = 3¶

Constant = 4¶

Local = 5¶

Tensor = 6¶

SharedCluster = 7¶

__str__()¶: Return str(self).

class mlir.dialects._nvvm_enum_gen.PrefetchCacheLevel¶

Bases: enum.IntEnum

NVVM Prefetch Cache Level

L1 = 0¶

L2 = 1¶

__str__()¶: Return str(self).

class mlir.dialects._nvvm_enum_gen.ProxyKind¶

Bases: enum.IntEnum

Proxy kind

alias = 0¶

async_ = 1¶

async_global = 2¶

async_shared = 3¶

TENSORMAP = 4¶

GENERIC = 5¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._proxykind(x, context)¶

class mlir.dialects._nvvm_enum_gen.ReduxKind¶

Bases: enum.IntEnum

NVVM redux kind

ADD = 1¶

AND = 2¶

MAX = 3¶

MIN = 4¶

OR = 5¶

UMAX = 6¶

UMIN = 7¶

XOR = 8¶

FMIN = 9¶

FMAX = 10¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._reduxkind(x, context)¶

class mlir.dialects._nvvm_enum_gen.SaturationMode¶

Bases: enum.IntEnum

NVVM SaturationMode kind

NONE = 0¶

SATFINITE = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._saturationmode(x, context)¶

class mlir.dialects._nvvm_enum_gen.SetMaxRegisterAction¶

Bases: enum.IntEnum

NVVM set max register action

decrease = 1¶

increase = 0¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._setmaxregisteraction(x, context)¶

class mlir.dialects._nvvm_enum_gen.SharedSpace¶

Bases: enum.IntEnum

Shared memory space

shared_cta = 0¶

shared_cluster = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._sharedspace(x, context)¶

class mlir.dialects._nvvm_enum_gen.ShflKind¶

Bases: enum.IntEnum

NVVM shuffle kind

bfly = 0¶

up = 1¶

down = 2¶

idx = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._shflkind(x, context)¶

class mlir.dialects._nvvm_enum_gen.TMALoadMode¶

Bases: enum.IntEnum

NVVM TMA Load Mode

TILE = 0¶

IM2COL = 1¶

IM2COL_W = 2¶

IM2COL_W_128 = 3¶

TILE_GATHER4 = 4¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tmaloadmode(x, context)¶

class mlir.dialects._nvvm_enum_gen.TMAReduxKind¶

Bases: enum.IntEnum

NVVM TMA redux kind

ADD = 0¶

MAX = 2¶

MIN = 1¶

INC = 3¶

DEC = 4¶

AND = 5¶

OR = 6¶

XOR = 7¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tmareduxkind(x, context)¶

class mlir.dialects._nvvm_enum_gen.TMAStoreMode¶

Bases: enum.IntEnum

NVVM TMA Store Mode

TILE = 0¶

IM2COL = 1¶

TILE_SCATTER4 = 2¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tmastoremode(x, context)¶

class mlir.dialects._nvvm_enum_gen.Tcgen05CpMulticast¶

Bases: enum.IntEnum

tcgen05 cp multicast

NONE = 0¶

WARPX2_02_13 = 1¶

WARPX2_01_23 = 2¶

WARPX4 = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tcgen05cpmulticast(x, context)¶

class mlir.dialects._nvvm_enum_gen.Tcgen05CpShape¶

Bases: enum.IntEnum

tcgen05 cp shapes

SHAPE_128x256b = 0¶

SHAPE_4x256b = 1¶

SHAPE_128x128b = 2¶

SHAPE_64x128b = 3¶

SHAPE_32x128b = 4¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tcgen05cpshape(x, context)¶

class mlir.dialects._nvvm_enum_gen.Tcgen05CpSrcFormat¶

Bases: enum.IntEnum

tcgen05 cp source format

B6x16_P32 = 0¶

B4x16_P64 = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tcgen05cpsrcformat(x, context)¶

class mlir.dialects._nvvm_enum_gen.Tcgen05FenceKind¶

Bases: enum.IntEnum

NVVM Tcgen05 fence kind

BEFORE_THREAD_SYNC = 0¶

AFTER_THREAD_SYNC = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tcgen05fencekind(x, context)¶

class mlir.dialects._nvvm_enum_gen.Tcgen05LdStShape¶

Bases: enum.IntEnum

allowed 32-bit signless integer cases: 0, 1, 2, 3, 4

SHAPE_16X64B = 0¶

SHAPE_16X128B = 1¶

SHAPE_16X256B = 2¶

SHAPE_32X32B = 3¶

SHAPE_16X32BX2 = 4¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tcgen05ldstshape(x, context)¶

class mlir.dialects._nvvm_enum_gen.Tcgen05WaitKind¶

Bases: enum.IntEnum

NVVM Tcgen05 wait kind

LOAD = 0¶

STORE = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._tcgen05waitkind(x, context)¶

class mlir.dialects._nvvm_enum_gen.VoteSyncKind¶

Bases: enum.IntEnum

NVVM vote sync kind

any = 0¶

all = 1¶

ballot = 2¶

uni = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._votesynckind(x, context)¶

class mlir.dialects._nvvm_enum_gen.WGMMAScaleIn¶

Bases: enum.IntEnum

WGMMA overflow options

one = 1¶

neg¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._wgmmascalein(x, context)¶

class mlir.dialects._nvvm_enum_gen.WGMMAScaleOut¶

Bases: enum.IntEnum

WGMMA input predicate

zero = 0¶

one = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._wgmmascaleout(x, context)¶

class mlir.dialects._nvvm_enum_gen.WGMMATypes¶

Bases: enum.IntEnum

NVVM WGMMA types

f16 = 0¶

tf32 = 1¶

u8 = 2¶

s8 = 3¶

b1 = 4¶

bf16 = 5¶

e4m3 = 6¶

e5m2 = 7¶

f32 = 8¶

s32 = 9¶

__str__()¶: Return str(self).

mlir.dialects._nvvm_enum_gen._wgmmatypes(x, context)¶

mlir.dialects._nvvm_enum_gen._barrierreductionattr(x, context)¶

mlir.dialects._nvvm_enum_gen._ctagroupkindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._cacheevictionpriorityattr(x, context)¶

mlir.dialects._nvvm_enum_gen._clusterlaunchcontrolquerytypeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._dotaccumulatetypeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._fproundingmodeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._griddepactionattr(x, context)¶

mlir.dialects._nvvm_enum_gen._ldstmatrixelttypeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._loadcachemodifierattr(x, context)¶

mlir.dialects._nvvm_enum_gen._mmab1opattr(x, context)¶

mlir.dialects._nvvm_enum_gen._mmafragattr(x, context)¶

mlir.dialects._nvvm_enum_gen._mmaintoverflowattr(x, context)¶

mlir.dialects._nvvm_enum_gen._mmalayoutattr(x, context)¶

mlir.dialects._nvvm_enum_gen._mmatypesattr(x, context)¶

mlir.dialects._nvvm_enum_gen._matchsynckindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._memscopekindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._nvvmmemoryspaceattr(x, context)¶

mlir.dialects._nvvm_enum_gen._prefetchcachelevelattr(x, context)¶

mlir.dialects._nvvm_enum_gen._proxykindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._reduxkindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._saturationmodeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._setmaxregisteractionattr(x, context)¶

mlir.dialects._nvvm_enum_gen._sharedspaceattr(x, context)¶

mlir.dialects._nvvm_enum_gen._shflkindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tmaloadmodeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tmareduxkindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tmastoremodeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tcgen05cpmulticastattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tcgen05cpshapeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tcgen05cpsrcformatattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tcgen05fencekindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tcgen05ldstshapeattr(x, context)¶

mlir.dialects._nvvm_enum_gen._tcgen05waitkindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._votesynckindattr(x, context)¶

mlir.dialects._nvvm_enum_gen._wgmmascaleinattr(x, context)¶

mlir.dialects._nvvm_enum_gen._wgmmascaleoutattr(x, context)¶

mlir.dialects._nvvm_enum_gen._wgmmatypesattr(x, context)¶