Skip to content

Lower tt.load and tt.store cacheModifier and volatile attributes #5471

@etiotto

Description

@etiotto

The Triton dialect load and store instructions (tt.load, tt.store) can take attributes. For example the load instruction may have the volatile and/or the cache attributes:

   tt.load %6 {isVolatile = true} : tensor<256x!tt.ptr<f32>, #blocked0>
   tt.load %6 cacheModifier = ca : tensor<256x!tt.ptr<f32>, #blocked0>

Similarly for the store instruction. During code generation to LLVM IR, these attributes are currently ignored so we need to handle them.

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions