support for GB10 (DGX Spark)

#1
by dbuos - opened

Hi, is there any plan to support GB10 GPU (Spark DGX)?
Currently the kernel fails to load on that machine:

FileNotFoundError: Cannot find a build variant for this system in flashrt/gated-delta-attention (revision: 6c56ec615b35ecd513b46eb247dbdf56e7537f7c):

torch211-cxx11-cu128-x86_64-linux: CPU (x86_64) does not match system CPU (aarch64)
flashrt org

Hi! This will be addressed in the update. I’m currently working on updating this type of kernel, and these improvements will be included and packaged together with the update. Thanks!!

Sign up or log in to comment