Introducing OptimizedStack based on uint256 integers (#156)

* Introducing OptimizedStack based on uint256 integers. * Replacing big.int with uint256 for jump instruction. * Update state/runtime/evm/instructions.go Co-authored-by: Goran Rojovic <[email protected]> * Code review fixes. * opMStore is changed to use newly introduced WriteToSlice() function with unrolled loop. This allows CPU to execute multiple instructions in parallel providing >200% increased speed compared to uint256.WriteToSlice() implementation. If needed, can be additionally implemented for different array lengths. * Fixing linter errors. * setBytes to use uint256 instead of big.int. * Fixing linter errors. * Go mod tidy * Lint fix * Revert go version * Address comments * Remove pointer dereferencing and use value type reference --------- Co-authored-by: Goran Rojovic <[email protected]> Co-authored-by: Stefan Negovanović <[email protected]>
Ethernal-Tech · Mar 28, 2024 · 8ca41b1 · 8ca41b1
1 parent 307c7bb
commit 8ca41b1
Show file tree

Hide file tree

Showing 14 changed files with 1,040 additions and 766 deletions.
diff --git a/e2e-polybft/e2e/jsonrpc_test.go b/e2e-polybft/e2e/jsonrpc_test.go
@@ -226,7 +226,7 @@ func TestE2E_JsonRPC(t *testing.T) {
 		require.NoError(t, err)
 
 		resp, err := newEthClient.Call(&bladeRPC.CallMsg{
-			From: types.Address(acctZeroBalance.Address()),
+			From: acctZeroBalance.Address(),
 			To:   &target,
 			Data: input,
 		}, bladeRPC.LatestBlockNumber, nil)

diff --git a/e2e-polybft/e2e/storage_test.go b/e2e-polybft/e2e/storage_test.go
@@ -66,7 +66,7 @@ func TestE2E_Storage(t *testing.T) {
 			}
 
 			txn.SetFrom(sender.Address())
-			txn.SetTo((*types.Address)(&to))
+			txn.SetTo(&to)
 			txn.SetGas(21000)
 			txn.SetValue(big.NewInt(int64(i)))
 			txn.SetNonce(uint64(i))

diff --git a/e2e-polybft/e2e/txpool_test.go b/e2e-polybft/e2e/txpool_test.go
@@ -59,7 +59,7 @@ func TestE2E_TxPool_Transfer(t *testing.T) {
 			if i%2 == 0 {
 				txData = types.NewDynamicFeeTx(
 					types.WithFrom(sender.Address()),
-					types.WithTo((*types.Address)(&to)),
+					types.WithTo(&to),
 					types.WithGas(30000), // enough to send a transfer
 					types.WithValue(big.NewInt(int64(sendAmount))),
 					types.WithNonce(uint64(i)),
@@ -69,7 +69,7 @@ func TestE2E_TxPool_Transfer(t *testing.T) {
 			} else {
 				txData = types.NewLegacyTx(
 					types.WithFrom(sender.Address()),
-					types.WithTo((*types.Address)(&to)),
+					types.WithTo(&to),
 					types.WithGas(30000),
 					types.WithValue(big.NewInt(int64(sendAmount))),
 					types.WithNonce(uint64(i)),

diff --git a/go.mod b/go.mod
@@ -53,6 +53,8 @@ require (
 	pgregory.net/rapid v1.1.0
 )
 
+require github.com/holiman/uint256 v1.2.4
+
 require (
 	cloud.google.com/go/compute v1.23.3 // indirect
 	cloud.google.com/go/compute/metadata v0.2.3 // indirect

diff --git a/go.sum b/go.sum
@@ -311,6 +311,8 @@ github.com/hashicorp/hcl v1.0.1-vault-5 h1:kI3hhbbyzr4dldA8UdTb7ZlVVlI2DACdCfz31
 github.com/hashicorp/hcl v1.0.1-vault-5/go.mod h1:XYhtn6ijBSAj6n4YqAaf7RBPS4I06AItNorpy+MoQNM=
 github.com/hashicorp/vault/api v1.12.0 h1:meCpJSesvzQyao8FCOgk2fGdoADAnbDu2WPJN1lDLJ4=
 github.com/hashicorp/vault/api v1.12.0/go.mod h1:si+lJCYO7oGkIoNPAN8j3azBLTn9SjMGS+jFaHd1Cck=
+github.com/holiman/uint256 v1.2.4 h1:jUc4Nk8fm9jZabQuqr2JzednajVmBpC+oiTiXZJEApU=
+github.com/holiman/uint256 v1.2.4/go.mod h1:EOMSn4q6Nyt9P6efbI3bueV4e1b3dGlUCXeiRV4ng7E=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
 github.com/huin/goupnp v1.3.0 h1:UvLUlWDNpoUdYzb2TCn+MuTWtcjXKSza2n6CBdQ0xXc=
 github.com/huin/goupnp v1.3.0/go.mod h1:gnGPsThkYa7bFi/KWmEysQRf48l2dvR5bxr2OFckNX8=

diff --git a/state/runtime/evm/dispatch_table_test.go b/state/runtime/evm/dispatch_table_test.go
@@ -25,11 +25,10 @@ func TestPushOpcodes(t *testing.T) {
 
 		assert.False(t, s.stop)
 
-		res := s.pop().Bytes()
-		assert.Len(t, res, c)
-
-		assert.True(t, bytes.HasPrefix(code[1:], res))
+		v := s.pop()
+		assert.Len(t, v.Bytes(), c)
 
+		assert.True(t, bytes.HasPrefix(code[1:], v.Bytes()))
 		c++
 	}
 }
diff --git a/state/runtime/evm/evm_test.go b/state/runtime/evm/evm_test.go
@@ -8,6 +8,7 @@ import (
 	"github.com/0xPolygon/polygon-edge/state/runtime"
 	"github.com/0xPolygon/polygon-edge/state/runtime/tracer"
 	"github.com/0xPolygon/polygon-edge/types"
+	"github.com/holiman/uint256"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 )
@@ -425,7 +426,7 @@ func TestRunWithTracer(t *testing.T) {
 			state.config = config
 
 			// make sure stack, memory, and returnData are empty
-			state.stack = make([]*big.Int, 0)
+			state.stack.data = make([]uint256.Int, 0)
 			state.memory = make([]byte, 0)
 			state.returnData = make([]byte, 0)