runtime: Float, Time
This commit is contained in:
@@ -25,9 +25,9 @@ func main() {
|
|||||||
var prompt *c.Char = c.Str("Once upon a time")
|
var prompt *c.Char = c.Str("Once upon a time")
|
||||||
var checkpointPath *c.Char = c.Str("stories15M.bin")
|
var checkpointPath *c.Char = c.Str("stories15M.bin")
|
||||||
var tokenizerPath *c.Char = c.Str("tokenizer.bin")
|
var tokenizerPath *c.Char = c.Str("tokenizer.bin")
|
||||||
var temperature, topp llama2.Float = 1.0, 0.9
|
var temperature, topp c.Float = 1.0, 0.9
|
||||||
var steps c.Int = 256
|
var steps c.Int = 256
|
||||||
var rngSeed uint64 = uint64(llama2.Time(nil))
|
var rngSeed uint64 = uint64(c.Time(nil))
|
||||||
|
|
||||||
// build the Transformer via the model .bin file
|
// build the Transformer via the model .bin file
|
||||||
var transformer llama2.Transformer
|
var transformer llama2.Transformer
|
||||||
|
|||||||
4
c/c.go
4
c/c.go
@@ -28,6 +28,7 @@ type (
|
|||||||
Char = int8
|
Char = int8
|
||||||
Int = C.int
|
Int = C.int
|
||||||
Uint = C.uint
|
Uint = C.uint
|
||||||
|
Float = float32
|
||||||
Pointer = unsafe.Pointer
|
Pointer = unsafe.Pointer
|
||||||
FilePtr = unsafe.Pointer
|
FilePtr = unsafe.Pointer
|
||||||
)
|
)
|
||||||
@@ -76,3 +77,6 @@ func Fprintf(fp FilePtr, format *Char, __llgo_va_list ...any) Int
|
|||||||
|
|
||||||
//go:linkname Qsort C.qsort
|
//go:linkname Qsort C.qsort
|
||||||
func Qsort(base Pointer, count, elem uintptr, compar func(a, b Pointer) Int)
|
func Qsort(base Pointer, count, elem uintptr, compar func(a, b Pointer) Int)
|
||||||
|
|
||||||
|
//go:linkname Time C.time
|
||||||
|
func Time(*int32) int32
|
||||||
|
|||||||
@@ -27,15 +27,12 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
Char = int8
|
Char = c.Char
|
||||||
Int = c.Int
|
Int = c.Int
|
||||||
Uint = c.Uint
|
Uint = c.Uint
|
||||||
Float = float32
|
Float = c.Float
|
||||||
)
|
)
|
||||||
|
|
||||||
//go:linkname Time C.time
|
|
||||||
func Time(*int32) int32
|
|
||||||
|
|
||||||
// -----------------------------------------------------------------------------
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
|||||||
Reference in New Issue
Block a user