mirror of
https://github.com/pseXperiments/icicle.git
synced 2026-01-07 22:53:56 -05:00
fb52650bbc
adds CI checks for building and testing Golang bindings adds CI checks for formatting Rust and Golang files Fixes Golang tests for BN254 Splits Actions checks for PR against main into multiple files Resolves #108 Resolves #107 Resolves #138
149 lines
4.0 KiB
Go
149 lines
4.0 KiB
Go
// Copyright 2023 Ingonyama
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// Code generated by Ingonyama DO NOT EDIT
|
|
|
|
package bn254
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/stretchr/testify/assert"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestNttBN254Batch(t *testing.T) {
|
|
count := 1 << 20
|
|
scalars := GenerateScalars(count, false)
|
|
|
|
nttResult := make([]G1ScalarField, len(scalars)) // Make a new slice with the same length
|
|
copy(nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, scalars)
|
|
NttBatch(&nttResult, false, count, 0)
|
|
assert.NotEqual(t, nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, nttResult)
|
|
}
|
|
|
|
func TestNttBN254CompareToGnarkDIF(t *testing.T) {
|
|
count := 1 << 2
|
|
scalars := GenerateScalars(count, false)
|
|
|
|
nttResult := make([]G1ScalarField, len(scalars)) // Make a new slice with the same length
|
|
copy(nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, scalars)
|
|
Ntt(&nttResult, false, 0)
|
|
assert.NotEqual(t, nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, nttResult)
|
|
}
|
|
|
|
func TestINttBN254CompareToGnarkDIT(t *testing.T) {
|
|
count := 1 << 3
|
|
scalars := GenerateScalars(count, false)
|
|
|
|
nttResult := make([]G1ScalarField, len(scalars)) // Make a new slice with the same length
|
|
copy(nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, scalars)
|
|
Ntt(&nttResult, true, 0)
|
|
assert.NotEqual(t, nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, nttResult)
|
|
}
|
|
|
|
func TestNttBN254(t *testing.T) {
|
|
count := 1 << 3
|
|
|
|
scalars := GenerateScalars(count, false)
|
|
|
|
nttResult := make([]G1ScalarField, len(scalars)) // Make a new slice with the same length
|
|
copy(nttResult, scalars)
|
|
|
|
assert.Equal(t, nttResult, scalars)
|
|
Ntt(&nttResult, false, 0)
|
|
assert.NotEqual(t, nttResult, scalars)
|
|
|
|
inttResult := make([]G1ScalarField, len(nttResult))
|
|
copy(inttResult, nttResult)
|
|
|
|
assert.Equal(t, inttResult, nttResult)
|
|
Ntt(&inttResult, true, 0)
|
|
assert.Equal(t, inttResult, scalars)
|
|
}
|
|
|
|
func TestNttBatchBN254(t *testing.T) {
|
|
count := 1 << 5
|
|
batches := 4
|
|
|
|
scalars := GenerateScalars(count*batches, false)
|
|
|
|
var scalarVecOfVec [][]G1ScalarField = make([][]G1ScalarField, 0)
|
|
|
|
for i := 0; i < batches; i++ {
|
|
start := i * count
|
|
end := (i + 1) * count
|
|
batch := make([]G1ScalarField, len(scalars[start:end]))
|
|
copy(batch, scalars[start:end])
|
|
scalarVecOfVec = append(scalarVecOfVec, batch)
|
|
}
|
|
|
|
nttBatchResult := make([]G1ScalarField, len(scalars))
|
|
copy(nttBatchResult, scalars)
|
|
|
|
NttBatch(&nttBatchResult, false, count, 0)
|
|
|
|
var nttResultVecOfVec [][]G1ScalarField
|
|
|
|
for i := 0; i < batches; i++ {
|
|
// Clone the slice
|
|
clone := make([]G1ScalarField, len(scalarVecOfVec[i]))
|
|
copy(clone, scalarVecOfVec[i])
|
|
|
|
// Add it to the result vector of vectors
|
|
nttResultVecOfVec = append(nttResultVecOfVec, clone)
|
|
|
|
// Call the ntt_bn254 function
|
|
Ntt(&nttResultVecOfVec[i], false, 0)
|
|
}
|
|
|
|
assert.NotEqual(t, nttBatchResult, scalars)
|
|
|
|
// Check that the ntt of each vec of scalars is equal to the intt of the specific batch
|
|
for i := 0; i < batches; i++ {
|
|
if !reflect.DeepEqual(nttResultVecOfVec[i], nttBatchResult[i*count:((i+1)*count)]) {
|
|
t.Errorf("ntt of vec of scalars not equal to intt of specific batch")
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkNTT(b *testing.B) {
|
|
LOG_NTT_SIZES := []int{12, 15, 20, 21, 22, 23, 24, 25, 26}
|
|
|
|
for _, logNTTSize := range LOG_NTT_SIZES {
|
|
nttSize := 1 << logNTTSize
|
|
b.Run(fmt.Sprintf("NTT %d", logNTTSize), func(b *testing.B) {
|
|
scalars := GenerateScalars(nttSize, false)
|
|
|
|
nttResult := make([]G1ScalarField, len(scalars)) // Make a new slice with the same length
|
|
copy(nttResult, scalars)
|
|
for n := 0; n < b.N; n++ {
|
|
Ntt(&nttResult, false, 0)
|
|
}
|
|
})
|
|
}
|
|
}
|