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

Integrate LLVM at llvm/llvm-project@8c5d9c79b9

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

PiperOrigin-RevId: 642352474
This commit is contained in:
jax authors 2024-06-11 12:23:56 -07:00 committed by jax authors
parent 2f749dbe39
commit d20b9e324f
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jaxlib/mosaic/gpu/passes.cc
View File

@ -56,8 +56,7 @@ class ConvertGpuToLLVMPass
converter.isLegal(op->getResultTypes());
});
auto symtab = mlir::SymbolTable(getOperation());
mlir::populateGpuToLLVMConversionPatterns(converter, patterns, "gpu.binary",
false, &symtab);
mlir::populateGpuToLLVMConversionPatterns(converter, patterns, false);
if (mlir::applyPartialConversion(getOperation(), target,
std::move(patterns))
.failed()) {
@ -75,4 +74,4 @@ void registerConvertGpuToLLVMPass() {
}
} // namespace gpu
} // namespace mosaic
} // namespace mosaic