mirror of
https://github.com/luxfi/accel.git
synced 2026-10-30 17:03:45 +00:00
zeekay
29de08c64c
Replace the single hardcoded `#cgo CFLAGS: -I${SRCDIR}/../../../mlx/include`
directive in ops/code/code_cpu.go with a multi-prefix probe that lists every
standard install location. The C compiler silently skips -I/-L paths that
don't exist, so the linker resolves whichever prefix actually has the
artefacts.
Fallback chain (in compiler search order):
- CGO_CFLAGS / CGO_LDFLAGS env override
- pkg-config lux-gpu
- /opt/homebrew/opt/lux-gpu (keg-only)
- /usr/local/opt/lux-gpu (Intel-Mac Homebrew)
- /opt/homebrew (Apple Silicon Homebrew)
- /usr/local (POSIX system install, cmake default)
- /opt/lux (Lux canonical prefix)
- \${SRCDIR}/../../../mlx (in-tree workspace dev fallback)
Switch crypto_secp256k1_cgo.go and ops/crypto/crypto_msm_native.go to
pkg-config (lux-crypto-secp256k1, lux-crypto-msm) so the same one-and-only-
way pattern applies across all CGO surfaces.
Add cgo_discovery.go + cgo_discovery_test.go exposing Provenance{}.GPUPaths()
so callers can introspect which prefix resolved at runtime. Tests pass
without any env vars set.
75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
// Copyright (C) 2024-2026, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
//go:build cgo && lux_crypto_native
|
|
// +build cgo,lux_crypto_native
|
|
|
|
package crypto
|
|
|
|
// msmCPU routes through luxcpp/crypto's gpukit_multi_pippenger_cpu C-ABI
|
|
// for all four wired curves (secp256k1, bn254 g1, bls12-381 g1, banderwagon).
|
|
// The same signed-digit Bos-Coster Pippenger body backs all of them.
|
|
//
|
|
// Build with `-tags=lux_crypto_native` to opt in. Linkage goes through the
|
|
// lux-crypto-msm pkg-config bundle, which exposes the gpukit + secp256k1 +
|
|
// bn254 + banderwagon + keccak CPU archives from $LUXCPP_PREFIX/lib.
|
|
//
|
|
// cmake -S $HOME/work/luxcpp/crypto -B $HOME/work/luxcpp/crypto/build \
|
|
// -DCMAKE_INSTALL_PREFIX=$HOME/work/luxcpp/install
|
|
// cmake --build $HOME/work/luxcpp/crypto/build \
|
|
// --target secp256k1_cpu keccak_cpu bn254_cpu banderwagon_cpu \
|
|
// gpukit_cpu gpukit_multi_pippenger
|
|
// cmake --install $HOME/work/luxcpp/crypto/build
|
|
//
|
|
// Then `export PKG_CONFIG_PATH=$HOME/work/luxcpp/install/lib/pkgconfig`.
|
|
|
|
/*
|
|
#cgo pkg-config: lux-crypto-msm
|
|
|
|
#include <lux/gpukit/multi_pippenger.h>
|
|
#include <lux/gpukit/gpukit.h>
|
|
*/
|
|
import "C"
|
|
|
|
import (
|
|
"unsafe"
|
|
)
|
|
|
|
func msmCPU(curve Curve, scalars, points [][]byte) ([]byte, error) {
|
|
n := len(scalars)
|
|
if n == 0 || n != len(points) {
|
|
return nil, ErrInvalidInput
|
|
}
|
|
if curve > CurveBanderwagon {
|
|
return nil, ErrUnsupported
|
|
}
|
|
const scalarSize = 32
|
|
pointSize := curve.pointSize()
|
|
|
|
scalarBuf := make([]byte, n*scalarSize)
|
|
pointBuf := make([]byte, n*pointSize)
|
|
for i := 0; i < n; i++ {
|
|
if len(scalars[i]) != scalarSize {
|
|
return nil, ErrInvalidInput
|
|
}
|
|
if len(points[i]) != pointSize {
|
|
return nil, ErrInvalidInput
|
|
}
|
|
copy(scalarBuf[i*scalarSize:], scalars[i])
|
|
copy(pointBuf[i*pointSize:], points[i])
|
|
}
|
|
|
|
out := make([]byte, pointSize)
|
|
st := C.gpukit_multi_pippenger_cpu(
|
|
C.uint32_t(curve),
|
|
(*C.uint8_t)(unsafe.Pointer(&scalarBuf[0])),
|
|
(*C.uint8_t)(unsafe.Pointer(&pointBuf[0])),
|
|
C.size_t(n),
|
|
(*C.uint8_t)(unsafe.Pointer(&out[0])),
|
|
)
|
|
if st != C.GPUKIT_OK {
|
|
return nil, ErrVerifyFailed
|
|
}
|
|
return out, nil
|
|
}
|