https://github.com/Wenyueh/MinivLLM/blob/main/src/myvllm/layers/linear.py Add test code in the case of a single machine with multiple GPUs.