Support for Intel AMX
Hi, just stumbled across the mojo language. Working in the AI field and looking to incorporate performance enhancements in my workflow using AVX-512 and Intel AMX instructions.
Noticed that mojo has an explicit SIMD type. Does this work with AMX instructions? If so, could you point me to some code examples?
If not, is this something that will be added to the language in the future?
2 Replies
AMX uses separate registers, so it needs to be dealt with separately. The built in SIMD does not support AMX since you can't use those registers as normal SIMD registers.
ok. Thanks!