tacco icon indicating copy to clipboard operation
tacco copied to clipboard

TACCO for ARM-based systems

Open LeonHafner opened this issue 11 months ago • 3 comments

Hi, thanks for the great tool. I wanted to asked if there are any plans to release a version of TACCO that works with the ARM architecture (for linux and apple silicon). Currently only Intel architectures are supported, as the mkl and mkl-service dependencies are exclusively available for those.

LeonHafner avatar Jan 08 '25 12:01 LeonHafner

Hi, thanks for pointing this out! It would indeed be neat to support ARM. From the code the changes seem to be minimal: the only essential mkl piece already has a non-mkl workaround https://github.com/simonwm/tacco/blob/87252c6ffcf5f616ffbf167a8d4d34bb8d86d276/tacco/utils/_math.py#L270 We mostly need to check how to best modify the (conda) dependencies such that those with x86 CPUs still get the best performance available on their system. Until then, you could hack it yourself or maybe use some x86 emulation?

simonwm avatar Jan 08 '25 19:01 simonwm

Great, it sounds like it shouldn't be too much work. Yes, no worries—I’ll run it on an x86 Linux machine in the meantime.

LeonHafner avatar Jan 10 '25 13:01 LeonHafner

Hi @LeonHafner

we are thinking about an ARM release currently. If you want to use it in the meantime on an Apple Silicon machine, you can try the conda-Rosetta2 setup: https://conda-forge.org/docs/user/tipsandtricks/#installing-apple-intel-packages-on-apple-silicon

We have tested it on an Apple M1 and it worked rightaway.

JWatter avatar Jan 24 '25 12:01 JWatter