github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BACKEND] Add store cache modifiers (#1826)

Browse Source 
Plumb through store cache modifiers.
This commit is contained in:
Thomas
2023-06-23 09:29:10 -07:00
committed by GitHub
parent 2eb7bc4b4c
commit 3d1cd89b54
5 changed files with 56 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Conversion/TritonGPUToLLVM/LoadStoreOpToLLVM.cpp
View File

@@ -354,6 +354,9 @@ struct StoreOpConversion
auto &ptxStoreInstr =
ptxBuilder.create<>("st")
->global()
.o("wb", op.getCache() == triton::CacheModifier::WB)
.o("cg", op.getCache() == triton::CacheModifier::CG)
.o("cs", op.getCache() == triton::CacheModifier::CS)
.o("L1::evict_first",
op.getEvict() == triton::EvictionPolicy::EVICT_FIRST)
.o("L1::evict_last",