Closed phoenix-meadowlark closed 1 year ago
It's maybe worth noting that this segfault happens before iree-compile
could let me know that llvmcpu
isn't a registered target backend (but llvm-cpu
is). If that validation happens before compilation then that might narrow things down.
This failure appears to occur before iree-stream-schedule-execution
We are seeing the same behavior in https://github.com/openxla/iree/issues/13459 so likely this is a dupe. We should swing back around when it is closed.
@rsuderman #13459 is closed - was this a duplicate? @silvasean for visibility
Verified that this compiles successfully now.
What happened?
Attempting to compile the training step dumped by PAX for a 1.3B GPT-3-like model causes
iree-compile
to produce the following segmentation fault:Steps to reproduce your issue
Download
jax_ir41_pmap__wrapped_step_fn.mlir
and runiree-compile
(tested to fail onllvmcpu
andcuda
):(It's being pmap'd to a single device so that shouldn't be relevant).
What component(s) does this issue relate to?
Compiler
Version information
7a0208a3a8385543a37e939c09db45bab151d83b
Additional context
No response