Merge branch 'main' of https://github.com/getumen/cuml into update-cuml

getumen · Aug 6, 2024 · 79046d7 · 79046d7
2 parents 26a2b97 + e5763e7
commit 79046d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/go/fil.go b/go/fil.go
@@ -152,7 +152,7 @@ func (m *FILModel) PredictSingleClassScore(
 
 	result := make([]float32, numRow)
 	for i := 0; i < numRow; i++ {
-		result[i] = resultRaw[i*m.NumClass()+1]
+		result[i] = resultRaw[i*2+1]
 	}
 	return result, nil
 }
@@ -165,6 +165,3 @@ func (m *FILModel) Close() error {
 	return err
 }
 
-func (m *FILModel) NumClass() int {
-	return m.raw.NumClass()
-}
diff --git a/go/rawcuml4go/device_memory.go b/go/rawcuml4go/device_memory.go
@@ -70,7 +70,7 @@ func UseArenaMemoryResource(arena_size uint64) (
 	error,
 ) {
 	var pointer C.DeviceMemoryResource
-	ret := C.UseArenaMemoryResource(&pointer, arena_size)
+	ret := C.UseArenaMemoryResource(&pointer, (C.size_t)(arena_size))
 	if ret != 0 {
 		return nil, ErrGetDeviceMemoryResource
 	}