mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Integrate LLVM at llvm/llvm-project@d33e33fde7

Browse Source 
Updates LLVM usage to match
[d33e33fde770](https://github.com/llvm/llvm-project/commit/d33e33fde770)

PiperOrigin-RevId: 718414171
This commit is contained in:
jax authors 2025-01-22 09:21:19 -08:00
parent 5c6b6c124f
commit 6c76cc4e36
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/mosaic/dialect/gpu/mosaic_gpu.cc
View File

@ -316,7 +316,7 @@ llvm::FailureOr<WGMMALayout> GetWgmmaLayout(mlir::Location loc,
return emitError(loc, llvm::formatv(params...));
};
auto [strides, offset] = mlir::getStridesAndOffset(type);
auto [strides, offset] = type.getStridesAndOffset();
WGMMALayout layout = WGMMALayout::RowMajor;
if (strides[3] == 1) {