mlir.dialects._nvgpu_enum_gen

Attributes

_ods_ir

Classes

RcpRoundingMode	Rounding mode of rcp
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

_rcproundingmode(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._ods_ir

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.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)