Cannot lower pallas_call on platform: cuda

Jax框架JAX-Triton不可用：

image




参考地址：
MLIR translation rule for primitive 'pallas_call' not found for platform cuda #18275


image




参考地址：

Make Pallas/GPU easier to install #18603


image

posted on 2024-01-18 00:30  Angry_Panda  阅读(31)  评论(0)    收藏  举报

导航