export F_f32toa

bytedance · May 27, 2024 · ba7cd82 · ba7cd82
1 parent ff66e4f
commit ba7cd82
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 9 deletions.
diff --git a/encoder/encoder_amd64.go b/encoder/encoder_amd64.go
@@ -22,6 +22,10 @@ import (
     `github.com/bytedance/sonic/internal/encoder`
 )
 
+func init() {
+    encoder.ForceUseVM()
+}
+
 // EnableFallback indicates if encoder use fallback
 const EnableFallback = false
 

diff --git a/internal/encoder/vm/vm.go b/internal/encoder/vm/vm.go
@@ -39,13 +39,13 @@ var (
 	_T_encoding_TextMarshaler = rt.UnpackType(vars.EncodingTextMarshalerType)
 )
 
-func print_instr(buf []byte, pc int, op ir.Op, ins *ir.Instr) {
+func print_instr(buf []byte, pc int, op ir.Op, ins *ir.Instr, p unsafe.Pointer) {
 	if len(buf) > 20 {
 		fmt.Println(string(buf[len(buf)-20:]))
 	} else {
 		fmt.Println(string(buf))
 	}
-	fmt.Printf("pc %04d, op %v, ins %#v\n", pc, op, ins.Disassemble())
+	fmt.Printf("pc %04d, op %v, ins %#v, ptr: %x\n", pc, op, ins.Disassemble(), p)
 }
 
 func Execute(b *[]byte, p unsafe.Pointer, s *vars.Stack, flags uint64, prog *ir.Program) (error) {
@@ -65,7 +65,7 @@ func Execute(b *[]byte, p unsafe.Pointer, s *vars.Stack, flags uint64, prog *ir.
 		pc++
 		op := ins.Op()
 
-		// print_instr(buf, pc, op, ins)
+		// print_instr(buf, pc, op, ins, p)
 
 		switch op {
 		case ir.OP_goto:

diff --git a/internal/native/avx/f32toa.go b/internal/native/avx/f32toa.go
diff --git a/internal/native/avx2/f32toa.go b/internal/native/avx2/f32toa.go
diff --git a/internal/native/dispatch_amd64.go b/internal/native/dispatch_amd64.go
@@ -164,6 +164,7 @@ func useSSE() {
     S_f64toa      = sse.S_f64toa
     __F64toa      = sse.F_f64toa
     S_f32toa      = sse.S_f32toa
+    __F32toa      = sse.F_f32toa
     __F64toa      = sse.F_f64toa
     S_i64toa      = sse.S_i64toa
     __I64toa      = sse.F_i64toa
@@ -200,6 +201,7 @@ func useAVX() {
     S_f64toa      = avx.S_f64toa
     __F64toa      = avx.F_f64toa
     S_f32toa      = avx.S_f32toa
+    __F32toa      = avx.F_f32toa
     __F64toa      = avx.F_f64toa
     S_i64toa      = avx.S_i64toa
     __I64toa      = avx.F_i64toa
@@ -235,6 +237,7 @@ func useAVX2() {
     S_f64toa      = avx2.S_f64toa
     __F64toa      = avx2.F_f64toa
     S_f32toa      = avx2.S_f32toa
+    __F32toa      = avx2.F_f32toa
     __F64toa      = avx2.F_f64toa
     S_i64toa      = avx2.S_i64toa
     __I64toa      = avx2.F_i64toa

diff --git a/internal/native/sse/f32toa.go b/internal/native/sse/f32toa.go