SIMD độc lập nền tảng trong Go: Bước tiến mới cho lập trình hiệu năng cao

Go vừa công bố một bước tiến quan trọng trong việc hỗ trợ SIMD (Single Instruction Multiple Data) — kỹ thuật cho phép CPU xử lý nhiều dữ liệu song song trong một lệnh duy nhất. Với Go 1.26 và 1.27, ngôn ngữ này đã có những API thử nghiệm mở ra khả năng tăng tốc đáng kể cho các tác vụ tính toán nặng.

SIMD là gì và tại sao nó quan trọng?

SIMD là tính năng phần cứng có mặt trên hầu hết CPU hiện đại, cho phép thực hiện cùng một phép toán trên một vector dữ liệu chỉ trong một lệnh. Ví dụ, thay vì cộng từng cặp số float64 một, CPU có thể cộng 8 cặp cùng lúc.

Điều này mang lại lợi ích lớn cho nhiều lĩnh vực:

• Mã hóa và bảo mật: tăng tốc thuật toán mã hóa, kiểm tra CRC
• Xử lý dữ liệu: tính toán trên mảng lớn, phân tích số liệu
• Trí tuệ nhân tạo: tăng tốc các phép toán ma trận, xử lý tensor
• Thu gom rác: ngay cả bộ thu gom rác Green Tea của Go cũng dùng SIMD để quét bộ nhớ nhanh hơn

Trước đây, muốn dùng SIMD trong Go, lập trình viên buộc phải viết assembly — một công việc khó khăn và chỉ đáng làm với những đoạn mã cực kỳ quan trọng về hiệu năng. Điều này khiến phần lớn phần mềm Go bỏ phí tiềm năng của CPU.

Thách thức: mỗi nền tảng một kiểu

Vấn đề lớn nhất khi thiết kế API SIMD là sự khác biệt khổng lồ giữa các nền tảng. Không chỉ khác về tập lệnh, mà còn khác về cách biểu diễn vector:

• Kích thước vector cố định: wasm, PowerPC, s390x dùng 128 bit; amd64 hỗ trợ 128, 256 và 512 bit
• Kích thước thay đổi: RISC-V cho phép vector từ 128 đến 65536 bit; Arm64 có NEON (128 bit cố định) và SVE (128-2048 bit)
• Cách xử lý mặt nạ (mask): một số kiến trúc dùng bitmask vector, số khác dùng thanh ghi mặt nạ chuyên dụng
• Tập lệnh khác nhau: ví dụ, wasm thiếu phép so sánh cho vector số nguyên 64 bit

Go 1.26 đã ra mắt gói archsimd phụ thuộc kiến trúc cho amd64, và Go 1.27 bổ sung arm64 (NEON) cùng wasm. Tuy nhiên, việc viết mã đa nền tảng vẫn rất phức tạp.

Gói simd: Viết một lần, chạy mọi nơi

Điểm nhấn của Go 1.27 là gói simd thử nghiệm — giao diện SIMD hoàn toàn khả chuyển, không phụ thuộc nền tảng lẫn kích thước vector. Gói này lấy cảm hứng từ Highway của C++.

Mục tiêu là hỗ trợ mã "simd" viết một lần với hiệu năng gần bằng assembly, đồng thời cung cấp giả lập (emulation) tốt trên các nền tảng chưa hỗ trợ SIMD.

Cách tiếp cận của gói simd:

• Loại bỏ vector kích thước cố định khỏi hệ thống kiểu
• Chỉ hỗ trợ các phép toán nằm trong giao của mọi nền tảng
• Lấp đầy khoảng trống bằng giả lập hiệu quả dựa trên các lệnh SIMD khác

Các kiểu vector chỉ là những kiểu nguyên thủy viết hoa, số nhiều, ví dụ simd.Uint8s hay simd.Float32s. Vector được nạp từ và lưu vào slice:

func innerProduct(x, y []float32) float32 {
    var a simd.Float32s
    var i int
    for i = 0; i < len(x); i += a.Len() {
        a = a.Load(x[i:])
        a = a.Mul(a.Load(y[i:]))
        sum += a.ReduceSum()
    }
    // ...
}

Để dùng gói thử nghiệm này, lập trình viên chỉ cần đặt biến môi trường GOEXPERIMENT=simd.

Tự bổ sung phép toán còn thiếu

Nếu gói simd chưa hỗ trợ một phép toán nào đó, lập trình viên có thể tự triển khai cho từng kiến trúc bằng các hàm ToArch và FromArch. Ví dụ, bài viết minh họa cách thêm phương thức Int8s.OnesCount() — đếm số bit 1 trong mỗi phần tử — vốn chưa có trong Go 1.27:

• Trên amd64: dùng bảng tra cứu (lookup table) và lệnh PermuteOrZero cho AVX/AVX2, hoặc lệnh gốc cho AVX512
• Trên arm64 NEON và wasm: đơn giản hơn vì cả hai đều hỗ trợ OnesCount() gốc
• Trên nền tảng không có SIMD: dùng hàm giả lập chung OnesCountEmulated

Điều thú vị là các chuyển đổi giao diện và type switch trông có vẻ kém hiệu quả, nhưng trình biên dịch sẽ chuyên biệt hóa và tối ưu hóa chúng đi.

Giao của các API và giả lập phương thức

Chiến lược của Go là bắt đầu với những phép toán được hỗ trợ ở mọi nơi, gồm nạp, lưu, số học và so sánh. Những lỗ hổng còn lại được lấp bằng cách thêm giả lập vào các API archsimd theo kiến trúc.

Nhiều giả lập chỉ cần 2-3 lệnh. Ví dụ:

• Dịch bit theo từng phần tử được giả lập bằng dịch vector
• So sánh không dấu được xử lý bằng so sánh có dấu cộng hai phép XOR với hằng số

Với những lệnh phức tạp như carryless multiply (quan trọng cho mã hóa và CRC), Go cung cấp giả lập với thời gian chạy không phụ thuộc đầu vào — yếu tố then chốt để chống tấn công kênh bên (side-channel).

Đối với các phép như cộng cặp (horizontal addition), Go 1.28 sẽ cung cấp phép toán cấp cao hơn là sum reduction, giúp người dùng không phụ thuộc vào độ dài vector cụ thể.

Điều khiển bằng GODEBUG

Trên các nền tảng có hỗ trợ phần cứng, hành vi có thể điều chỉnh qua GODEBUG để dễ kiểm thử:

• GODEBUG=simd=0: dùng giả lập dù phần cứng có hỗ trợ
• GODEBUG=simd=128, =256, =512: dùng vector kích thước tương ứng, panic nếu tính năng không có
• GODEBUG=simd=+128, =+256, =+512: cho phép chạy dù thiếu một số tính năng, chỉ panic khi lệnh không hỗ trợ thực sự được dùng

Ví dụ điển hình là Raspberry Pi — hỗ trợ NEON nhưng thiếu PMULL (carryless multiply), hay Apple Silicon chạy giả lập amd64 — hỗ trợ AVX2 nhưng thiếu VPCLMULQDQ.

Chi tiết triển khai

Gói simd thực chất là sự kết hợp của một package công khai, một package triển khai nội bộ, và việc viết lại AST ở frontend trình biên dịch.

Trình biên dịch tạo ra nhiều bản sao chuyên biệt hóa của hàm, biến và kiểu có nhắc đến kiểu simd, thay thế chúng bằng các kiểu chuyên biệt trong simd/internal/bridge. Mỗi bản sao nhận hậu tố dạng @simdNNN với NNN là độ dài vector hoặc 0 (giả lập).

Cách này là thỏa hiệp giữa nhân bản mã và hiệu năng SIMD: chi phí điều phối được đẩy lên cao nhất có thể để tránh phải điều phối bên trong các phép tính SIMD.

Tương lai

Go 1.28 dự kiến sẽ:

• Bổ sung hỗ trợ SVE vào cả archsimd lẫn simd
• Thêm nhiều phép toán SIMD mới: OnesCount, thao tác mặt nạ, phép rút gọn, hoán vị vector
• Cung cấp biến thể tính năng để tránh phải giả lập hoàn toàn trên những nền tảng chỉ thiếu một vài lệnh

Đối với cộng đồng lập trình viên Go tại Việt Nam, đây là tín hiệu đáng mừng: các dự án xử lý dữ liệu lớn, ứng dụng AI, hay dịch vụ mã hóa sẽ có thể tận dụng sức mạnh phần cứng mà không phải viết assembly phức tạp. Những ai làm việc với Go trên các hệ thống nhúng, edge computing hay xử lý tín hiệu nên theo dõi sát gói thử nghiệm này trong các bản phát hành tới.