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@9d24f94379

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

PiperOrigin-RevId: 728265165
This commit is contained in:
jax authors 2025-02-18 10:28:21 -08:00
parent 82216fc40d
commit 725087e13f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
jaxlib/mosaic/gpu/custom_call.cc
View File

@ -130,15 +130,15 @@ mlir::FailureOr<mlir::OpPassManager> GetPassPipeline(
mlir::registerStripDebugInfo();
mlir::registerConvertNVGPUToNVVMPass();
mlir::registerConvertVectorToSCF();
mlir::registerSCFToControlFlow();
mlir::registerSCFToControlFlowPass();
mlir::registerConvertNVVMToLLVMPass();
mlir::registerArithToLLVMConversionPass();
mlir::registerConvertIndexToLLVMPass();
mlir::registerConvertGpuOpsToNVVMOps();
mlir::registerConvertMathToLLVMPass();
mlir::registerConvertFuncToLLVMPass();
mlir::registerConvertAffineToStandard();
mlir::registerReconcileUnrealizedCasts();
mlir::registerLowerAffinePass();
mlir::registerReconcileUnrealizedCastsPass();
// TODO(apaszke): Only register the passes we actually use.
mlir::memref::registerMemRefPasses();
mlir::registerConvertToLLVMPass();