mlir.dialects._nvgpu_enum_gen¶

Attributes¶

_ods_ir

Classes¶

`FPRoundingMode`	NVVM FPRoundingMode kind
`RcpRoundingMode`	Rounding mode of rcp
`SaturationMode`	NVVM SaturationMode kind
`TensorMapInterleaveKind`	Tensor map interleave layout type
`TensorMapL2PromoKind`	Tensor map L2 promotion type
`TensorMapOOBKind`	Tensor map out-of-bounds fill type
`TensorMapSwizzleKind`	Tensor map swizzling mode of shared memory banks

Functions¶

`register_attribute_builder`(kind[, replace, allow_existing])
`_fproundingmode`(x, context)
`_rcproundingmode`(x, context)
`_saturationmode`(x, context)
`_tensormapinterleavekind`(x, context)
`_tensormapl2promokind`(x, context)
`_tensormapoobkind`(x, context)
`_tensormapswizzlekind`(x, context)
`_rcproundingmodeattr`(x, context)
`_tensormapinterleaveattr`(x, context)
`_tensormapl2promoattr`(x, context)
`_tensormapoobattr`(x, context)
`_tensormapswizzleattr`(x, context)

Module Contents¶

mlir.dialects._nvgpu_enum_gen.register_attribute_builder(kind, replace=False, allow_existing=False)¶

mlir.dialects._nvgpu_enum_gen._ods_ir¶

class mlir.dialects._nvgpu_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._nvgpu_enum_gen._fproundingmode(x, context)¶

class mlir.dialects._nvgpu_enum_gen.RcpRoundingMode¶

Bases: enum.IntEnum

Rounding mode of rcp

APPROX = 0¶

RN = 1¶

RZ = 2¶

RM = 3¶

RP = 4¶

__str__()¶: Return str(self).

mlir.dialects._nvgpu_enum_gen._rcproundingmode(x, context)¶

class mlir.dialects._nvgpu_enum_gen.SaturationMode¶

Bases: enum.IntEnum

NVVM SaturationMode kind

NONE = 0¶

SATFINITE = 1¶

SAT = 2¶

__str__()¶: Return str(self).

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

class mlir.dialects._nvgpu_enum_gen.TensorMapInterleaveKind¶

Bases: enum.IntEnum

Tensor map interleave layout type

INTERLEAVE_NONE = 0¶

INTERLEAVE_16B = 1¶

INTERLEAVE_32B = 2¶

__str__()¶: Return str(self).

mlir.dialects._nvgpu_enum_gen._tensormapinterleavekind(x, context)¶

class mlir.dialects._nvgpu_enum_gen.TensorMapL2PromoKind¶

Bases: enum.IntEnum

Tensor map L2 promotion type

L2PROMO_NONE = 0¶

L2PROMO_64B = 1¶

L2PROMO_128B = 2¶

L2PROMO_256B = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvgpu_enum_gen._tensormapl2promokind(x, context)¶

class mlir.dialects._nvgpu_enum_gen.TensorMapOOBKind¶

Bases: enum.IntEnum

Tensor map out-of-bounds fill type

OOB_ZERO = 0¶

OOB_NAN = 1¶

__str__()¶: Return str(self).

mlir.dialects._nvgpu_enum_gen._tensormapoobkind(x, context)¶

class mlir.dialects._nvgpu_enum_gen.TensorMapSwizzleKind¶

Bases: enum.IntEnum

Tensor map swizzling mode of shared memory banks

SWIZZLE_NONE = 0¶

SWIZZLE_32B = 1¶

SWIZZLE_64B = 2¶

SWIZZLE_128B = 3¶

__str__()¶: Return str(self).

mlir.dialects._nvgpu_enum_gen._tensormapswizzlekind(x, context)¶

mlir.dialects._nvgpu_enum_gen._rcproundingmodeattr(x, context)¶

mlir.dialects._nvgpu_enum_gen._tensormapinterleaveattr(x, context)¶

mlir.dialects._nvgpu_enum_gen._tensormapl2promoattr(x, context)¶

mlir.dialects._nvgpu_enum_gen._tensormapoobattr(x, context)¶

mlir.dialects._nvgpu_enum_gen._tensormapswizzleattr(x, context)¶