triton-lang / triton

Development repository for the Triton language and compiler
https://triton-lang.org/
MIT License
12.53k stars 1.52k forks source link

Support L2 cache hint/policy #3438

Open manman-ren opened 5 months ago

manman-ren commented 5 months ago

Triton currently support L1 eviction policy but for L2 cache hint, it has the following TODO // Here lack a mlir::Value to bind to this operation, so disabled. // if (has_l2_evict_policy) https://github.com/openai/triton/blob/main/third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/LoadStoreOpToLLVM.cpp#L262

manman-ren commented 5 months ago

@ptillet As discussed in slack, I will take a look at this.