[Mlir-commits] [mlir] Reimplementing target description concept using DLTI attribute (PR #92138)

Renato Golin llvmlistbot at llvm.org
Thu May 16 07:51:12 PDT 2024


================
@@ -134,6 +136,24 @@ transposePackedMatmul(RewriterBase &rewriter, linalg::LinalgOp linalgOp,
   return packTransposedMatmul;
 }
 
+static SmallVector<int64_t> getDefaultBlockFactors(linalg::LinalgOp linalgOp) {
+  // get L1 cache size first.
+  uint32_t L1_cache_size = 4096; // default value
+  uint32_t cpuID = 0;
----------------
rengolin wrote:

I think it's just a naming issue, this should be `deviceID` or something. But also, I wouldn't assume the first one has it either. Perhaps a helper that, when not provided a device ID, returns a `pair<value, id>` if any device has it.

https://github.com/llvm/llvm-project/pull/92138


More information about the Mlir-commits mailing list