This ticket tracks further items of work relating to SIMD support in GHC native code generators (X86, AArch64, PPC), after the main branch of work lands (!12860), for the purposes of collaboration. These work items broadly correspond to remaining `SIMD NCG TODO` comments in the GHC codebase. Separate tickets will be created for the individual issues. Contributors interested in helping out should declare their interest in one of the following tasks, and they will be assigned the corresponding ticket. In roughly increasing order of difficulty: - [x] Add support for negation of integer vectors: change the `MO_VS_Neg` case in `GHC.CmmToAsm.X86.CodeGen.getRegister'`. - [x] Add support for extracting 32, 16 and 8 bit wide integer values from a vector: change the `MO_V_Extract` case in `GHC.CmmToAsm.X86.CodeGen.getRegister'`. Look at Godbolt output for Intel SIMD intrinsics such as `_mm_extract_epi8{8,16,32}`. - [x] Add support for inserting 32, 16 and 8 bit wide integer values into a vector: change `vector_int_insert_sse` in `GHC.CmmToAsm.X86.CodeGen`. Look at Godbolt output for Intel SIMD intrinsics such as `_mm_insert_epi{8,16,32}`. - [x] Add support for broadcasting 32, 16 and 8 bit wide integer values to vectors: change the `MO_V_Broadcast` case in `GHC.CmmToAsm.X86.CodeGen.getRegister'`. Look at Godbolt output for Intel SIMD intrinsics such as `_mm_set1_epi{8,16,32}`. - [x] Add support for arithmetic operations on integer vectors such as `MO_V_Add`, `MO_V_Mul`, `MO_VU_Quot` etc. This would involve changing `GHC.CmmToAsm.X86.CodeGen.getRegister'` to generate code for these `MachOp`s. (I expect that the main difficulty will be in handling all the different word sizes.) - [ ] Improve efficiency of vector pack instructions: instead of emitting a bunch of vector insertions in a row, it should be possible to write more efficient code. This would involve only changing `GHC.StgToCmm.Prim.doVecPackOp`. I would recommend looking at Godbolt output for Intel SSE intrinsics such as `_mm_set_ps` and `_mm_set_pd`. - [x] Improve the code we generate for shuffle primops. See `shuffleInstructions` in `GHC.CmmToAsm.X86.CodeGen`. - [x] Add support for integer vector shuffle operations, i.e. handling `MO_V_Shuffle` in `GHC.CmmToAsm.X86.CodeGen.getRegister'`. It will be a bit tricky to handle all of the cases, in particular shuffling 8 bit values. - [ ] Constant folding for vectors (see also #26915). This would involve (at least): - [ ] adding cases for vector primops in `GHC.Core.Opt.ConstantFold` - [ ] adding cases for vector `MachOp`s in `GHC.Cmm.Opt.cmmMachOpFoldM` - [x] updating `GHC.CmmToAsm.X86.CodeGen.getRegister' ... (CmmLit lit)` - [ ] Add support for 128-bit wide vectors in the PowerPC backend using the `VSX` instruction set. - Requires adding the instructions to `GHC.CmmToAsm.PPC.Instr`, updating `regUsageOfInstr` to store the more precise format that vector registers are used at, and making use of these instructions to implement the vector `MachOp`s. - Will require updating the testsuite's `get_cpu_features` function to check for availability of these instructions. - [x] Add support for 128-bit wide vectors in the AArch64 backend using the `NEON` instruction set. - Requires adding the instructions to `GHC.CmmToAsm.AArch64.Instr`, and updating `regUsageOfInstr` to store the more precise format that vector registers are used at, and making use of these instructions to implement the vector `MachOp`s. - Will require updating the testsuite's `get_cpu_features` function to check for availability of these instructions. - [ ] Add support for SIMD primops in test-primops. (This would first require adding floating-point primop support, which itself probably requires fixing the issues with floating point support in GHC outlined in #8364.) - [ ] Add support for SIMD registers in GHCi. - Requires careful consideration of functions such as `realArgRegsCover`, and probably finding a way of modifying `POP_ARG_REGS` so that we generate a jump which is annotated with exactly the right size of register, e.g. for passing YMM1 we want to annotate the jump with YMM1 (along with GP registers), not XMM1 or ZMM1. It will also be necessary to ensure that this Cmm is correctly compiled (e.g. passing the right flags to LLVM, as any compilation of e.g. YMM1 without `-mavx2` will lead to carnage). The bitset machinery in `mkNativeCallInfoSig` will also need to be updated. - [ ] Add support for ymm/zmm registers in the X86 NCG. The basic task would simply be to add new constructors to `VirtualReg` for 256 and 512 wide vectors and use those in `GHC.CmmToAsm.X86.RegInfo.mkVirtualReg`, but implementing code generation for all of the `MachOp`s (and adding corresponding tests), would constitute the bulk of the work.