<p dir="auto">This is a build at <code class="notranslate">jaxlib-v0.1.76</code> on a <code class="notranslate">ppc64le</code> machine. When running with GPUs, the following program executes without error. However, when running with CPUs, it fails. I have two questions, and thanks in advance for the help!</p>
<ol dir="auto">
<li>What is the cause of the error, and is there any way to fix it? Presumably, this is caused by the use of <code class="notranslate">ppc64le</code> CPUs, but I'm not super familiar with these.</li>
<li>I wouldn't mind having this issue as long as programs run using GPUs. However, would the following issue on the CPU side cause some (silent/subtle) issues when executing on GPUs? Put it in a different way, if I only use GPUs, do I need to worry about the error on the CPUs side?</li>
</ol>
<p dir="auto">Please:</p>
<ul class="contains-task-list">
<li class="task-list-item">
<p dir="auto"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Check for duplicate issues.</p>
</li>
<li class="task-list-item">
<p dir="auto"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox"> Provide a complete example of how to reproduce the bug, wrapped in triple backticks like this:</p>
</li>
<li class="task-list-item">
<p dir="auto"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> If applicable, include full error messages/tracebacks.</p>
</li>
</ul>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="(py38)&gt; ipython
Python 3.8.12 (default, Oct 12 2021, 13:02:29) 
Type 'copyright', 'credits' or 'license' for more information
IPython 7.31.1 -- An enhanced Interactive Python. Type '?' for help.

In [1]: import jax, jax.numpy as jnp

In [2]: jax.config.update(&quot;jax_platform_name&quot;, &quot;cpu&quot;)

In [3]: jnp.zeros([3])
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
LLVM ERROR: 64-bit code requested on a subtarget that doesn't support it!
Aborted (core dumped)"><pre class="notranslate"><code class="notranslate">(py38)&gt; ipython
Python 3.8.12 (default, Oct 12 2021, 13:02:29) 
Type 'copyright', 'credits' or 'license' for more information
IPython 7.31.1 -- An enhanced Interactive Python. Type '?' for help.

In [1]: import jax, jax.numpy as jnp

In [2]: jax.config.update("jax_platform_name", "cpu")

In [3]: jnp.zeros([3])
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
'pwr9' is not a recognized processor for this target (ignoring processor)
LLVM ERROR: 64-bit code requested on a subtarget that doesn't support it!
Aborted (core dumped)
</code></pre></div>
<p dir="auto">Please:</p>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Check for duplicate requests.</li>
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Describe your goal, and if possible provide a code snippet with a motivating example.</li>
</ul>
<hr>
<p dir="auto">The docs for <a href="https://jax.readthedocs.io/en/latest/_autosummary/jax.lax.dot_general.html" rel="nofollow"><code class="notranslate">jax.lax.dot_general</code></a> are unclear. In particular, the <code class="notranslate">dimension_numbers</code> argument isn't clear to me.</p>