'rocdl' Dialect

Operation definition ¶

rocdl.barrier (ROCDL::BarrierOp) ¶

Syntax:

operation ::= `rocdl.barrier` attr-dict

rocdl.buffer.load (ROCDL::MubufLoadOp) ¶

Operands: ¶

Results: ¶

rocdl.buffer.store (ROCDL::MubufStoreOp) ¶

Operands: ¶

rocdl.ds_bpermute (ROCDL::DsBpermuteOp) ¶

Syntax:

operation ::= `rocdl.ds_bpermute` $index `,` $src  attr-dict `:` `(` type($index) `,` type($src) `)` `->` type($res)

Operands: ¶

Results: ¶

rocdl.ds_swizzle (ROCDL::DsSwizzleOp) ¶

Syntax:

operation ::= `rocdl.ds_swizzle` $src `,` $offset  attr-dict `:` `(` type($src) `,` type($offset) `)` `->` type($res)

Operands: ¶

Results: ¶

rocdl.grid.dim.x (ROCDL::GridDimXOp) ¶

Syntax:

operation ::= `rocdl.grid.dim.x` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.grid.dim.y (ROCDL::GridDimYOp) ¶

Syntax:

operation ::= `rocdl.grid.dim.y` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.grid.dim.z (ROCDL::GridDimZOp) ¶

Syntax:

operation ::= `rocdl.grid.dim.z` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.mbcnt.hi (ROCDL::MbcntHiOp) ¶

Syntax:

operation ::= `rocdl.mbcnt.hi` $in0 `,` $in1  attr-dict `:` `(` type($in0) `,` type($in1) `)` `->` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands: ¶

Results: ¶

rocdl.mbcnt.lo (ROCDL::MbcntLoOp) ¶

Syntax:

operation ::= `rocdl.mbcnt.lo` $in0 `,` $in1  attr-dict `:` `(` type($in0) `,` type($in1) `)` `->` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x16bf16.1k (ROCDL::mfma_f32_16x16x16bf16_1k) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x16bf16.1k` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x16f16 (ROCDL::mfma_f32_16x16x16f16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x16f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x1f32 (ROCDL::mfma_f32_16x16x1f32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x1f32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x2bf16 (ROCDL::mfma_f32_16x16x2bf16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x2bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x32.bf8.bf8 (ROCDL::mfma_f32_16x16x32_bf8_bf8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x32.bf8.bf8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x32.bf8.fp8 (ROCDL::mfma_f32_16x16x32_bf8_fp8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x32.bf8.fp8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x32.fp8.bf8 (ROCDL::mfma_f32_16x16x32_fp8_bf8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x32.fp8.bf8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x32.fp8.fp8 (ROCDL::mfma_f32_16x16x32_fp8_fp8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x32.fp8.fp8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x4bf16.1k (ROCDL::mfma_f32_16x16x4bf16_1k) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x4bf16.1k` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x4f16 (ROCDL::mfma_f32_16x16x4f16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x4f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x4f32 (ROCDL::mfma_f32_16x16x4f32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x4f32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x8.xf32 (ROCDL::mfma_f32_16x16x8_xf32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x8.xf32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.16x16x8bf16 (ROCDL::mfma_f32_16x16x8bf16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.16x16x8bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x16.bf8.bf8 (ROCDL::mfma_f32_32x32x16_bf8_bf8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x16.bf8.bf8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x16.bf8.fp8 (ROCDL::mfma_f32_32x32x16_bf8_fp8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x16.bf8.fp8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x16.fp8.bf8 (ROCDL::mfma_f32_32x32x16_fp8_bf8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x16.fp8.bf8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x16.fp8.fp8 (ROCDL::mfma_f32_32x32x16_fp8_fp8) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x16.fp8.fp8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x1f32 (ROCDL::mfma_f32_32x32x1f32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x1f32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x2bf16 (ROCDL::mfma_f32_32x32x2bf16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x2bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x2f32 (ROCDL::mfma_f32_32x32x2f32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x2f32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x4.xf32 (ROCDL::mfma_f32_32x32x4_xf32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x4.xf32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x4bf16 (ROCDL::mfma_f32_32x32x4bf16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x4bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x4bf16.1k (ROCDL::mfma_f32_32x32x4bf16_1k) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x4bf16.1k` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x4f16 (ROCDL::mfma_f32_32x32x4f16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x4f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x8bf16.1k (ROCDL::mfma_f32_32x32x8bf16_1k) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x8bf16.1k` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.32x32x8f16 (ROCDL::mfma_f32_32x32x8f16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.32x32x8f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.4x4x1f32 (ROCDL::mfma_f32_4x4x1f32) ¶

Syntax:

operation ::= `rocdl.mfma.f32.4x4x1f32` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.4x4x2bf16 (ROCDL::mfma_f32_4x4x2bf16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.4x4x2bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.4x4x4bf16.1k (ROCDL::mfma_f32_4x4x4bf16_1k) ¶

Syntax:

operation ::= `rocdl.mfma.f32.4x4x4bf16.1k` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f32.4x4x4f16 (ROCDL::mfma_f32_4x4x4f16) ¶

Syntax:

operation ::= `rocdl.mfma.f32.4x4x4f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f64.16x16x4f64 (ROCDL::mfma_f64_16x16x4f64) ¶

Syntax:

operation ::= `rocdl.mfma.f64.16x16x4f64` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.f64.4x4x4f64 (ROCDL::mfma_f64_4x4x4f64) ¶

Syntax:

operation ::= `rocdl.mfma.f64.4x4x4f64` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.16x16x16i8 (ROCDL::mfma_i32_16x16x16i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.16x16x16i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.16x16x32.i8 (ROCDL::mfma_i32_16x16x32_i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.16x16x32.i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.16x16x4i8 (ROCDL::mfma_i32_16x16x4i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.16x16x4i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.32x32x16.i8 (ROCDL::mfma_i32_32x32x16_i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.32x32x16.i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.32x32x4i8 (ROCDL::mfma_i32_32x32x4i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.32x32x4i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.32x32x8i8 (ROCDL::mfma_i32_32x32x8i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.32x32x8i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.mfma.i32.4x4x4i8 (ROCDL::mfma_i32_4x4x4i8) ¶

Syntax:

operation ::= `rocdl.mfma.i32.4x4x4i8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.raw.buffer.atomic.cmpswap (ROCDL::RawBufferAtomicCmpSwap) ¶

Syntax:

operation ::= `rocdl.raw.buffer.atomic.cmpswap` attr-dict `(` operands `)` `:` type($res) `,` type($rsrc)

Operands: ¶

Results: ¶

rocdl.raw.buffer.atomic.fadd (ROCDL::RawBufferAtomicFAddOp) ¶

Operands: ¶

rocdl.raw.buffer.atomic.fmax (ROCDL::RawBufferAtomicFMaxOp) ¶

Operands: ¶

rocdl.raw.buffer.atomic.smax (ROCDL::RawBufferAtomicSMaxOp) ¶

Operands: ¶

rocdl.raw.buffer.atomic.umin (ROCDL::RawBufferAtomicUMinOp) ¶

Operands: ¶

rocdl.raw.buffer.load (ROCDL::RawBufferLoadOp) ¶

Operands: ¶

Results: ¶

rocdl.raw.buffer.store (ROCDL::RawBufferStoreOp) ¶

Operands: ¶

rocdl.wmma.bf16.16x16x16.bf16 (ROCDL::wmma_bf16_16x16x16_bf16) ¶

Syntax:

operation ::= `rocdl.wmma.bf16.16x16x16.bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.wmma.f16.16x16x16.f16 (ROCDL::wmma_f16_16x16x16_f16) ¶

Syntax:

operation ::= `rocdl.wmma.f16.16x16x16.f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.wmma.f32.16x16x16.bf16 (ROCDL::wmma_f32_16x16x16_bf16) ¶

Syntax:

operation ::= `rocdl.wmma.f32.16x16x16.bf16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.wmma.f32.16x16x16.f16 (ROCDL::wmma_f32_16x16x16_f16) ¶

Syntax:

operation ::= `rocdl.wmma.f32.16x16x16.f16` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.wmma.i32.16x16x16.iu4 (ROCDL::wmma_i32_16x16x16_iu4) ¶

Syntax:

operation ::= `rocdl.wmma.i32.16x16x16.iu4` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.wmma.i32.16x16x16.iu8 (ROCDL::wmma_i32_16x16x16_iu8) ¶

Syntax:

operation ::= `rocdl.wmma.i32.16x16x16.iu8` $args attr-dict `:` functional-type($args, $res)

Operands: ¶

Results: ¶

rocdl.workgroup.dim.x (ROCDL::BlockDimXOp) ¶

Syntax:

operation ::= `rocdl.workgroup.dim.x` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workgroup.dim.y (ROCDL::BlockDimYOp) ¶

Syntax:

operation ::= `rocdl.workgroup.dim.y` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workgroup.dim.z (ROCDL::BlockDimZOp) ¶

Syntax:

operation ::= `rocdl.workgroup.dim.z` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workgroup.id.x (ROCDL::BlockIdXOp) ¶

Syntax:

operation ::= `rocdl.workgroup.id.x` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workgroup.id.y (ROCDL::BlockIdYOp) ¶

Syntax:

operation ::= `rocdl.workgroup.id.y` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workgroup.id.z (ROCDL::BlockIdZOp) ¶

Syntax:

operation ::= `rocdl.workgroup.id.z` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workitem.id.x (ROCDL::ThreadIdXOp) ¶

Syntax:

operation ::= `rocdl.workitem.id.x` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workitem.id.y (ROCDL::ThreadIdYOp) ¶

Syntax:

operation ::= `rocdl.workitem.id.y` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

rocdl.workitem.id.z (ROCDL::ThreadIdZOp) ¶

Syntax:

operation ::= `rocdl.workitem.id.z` attr-dict `:` type($res)

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable, NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Results: ¶

Attribute definition ¶

ROCDLTargetAttr ¶

Syntax:

#rocdl.target<
  int,   # O
  ::llvm::StringRef,   # triple
  ::llvm::StringRef,   # chip
  ::llvm::StringRef,   # features
  ::llvm::StringRef,   # abi
  DictionaryAttr,   # flags
  ArrayAttr   # link
>

ROCDL target attribute for controlling compilation of AMDGPU targets. All parameters decay into default values if not present.

Examples:

1. Target with default values.

  gpu.module @mymodule [#rocdl.target] attributes {...} {
    ...
  }

2. Target with gfx90a chip and fast math.

  gpu.module @mymodule [#rocdl.target<chip = "gfx90a", flags = {fast, no_wave64}>] {
    ...
  }

Parameters: ¶