mirror of
https://github.com/pseXperiments/icicle.git
synced 2026-01-09 23:48:10 -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
50 lines
1011 B
Go
50 lines
1011 B
Go
package goicicle
|
|
|
|
// This file implements CUDA driver context management
|
|
|
|
// #cgo CFLAGS: -I /usr/local/cuda/include
|
|
// #cgo LDFLAGS: -L/usr/local/cuda/lib64 -lcudart
|
|
/*
|
|
#include <cuda.h>
|
|
#include <cuda_runtime.h>
|
|
*/
|
|
import "C"
|
|
|
|
import (
|
|
"errors"
|
|
"unsafe"
|
|
)
|
|
|
|
func CudaMalloc(size int) (dp unsafe.Pointer, err error) {
|
|
var p C.void
|
|
dp = unsafe.Pointer(&p)
|
|
if err := C.cudaMalloc(&dp, C.size_t(size)); err != 0 {
|
|
return nil, errors.New("could not create memory space")
|
|
}
|
|
return dp, nil
|
|
}
|
|
|
|
func CudaFree(dp unsafe.Pointer) int {
|
|
if err := C.cudaFree(dp); err != 0 {
|
|
return -1
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func CudaMemCpyHtoD[T any](dst_d unsafe.Pointer, src []T, size int) int {
|
|
src_c := unsafe.Pointer(&src[0])
|
|
if err := C.cudaMemcpy(dst_d, src_c, C.size_t(size), 1); err != 0 {
|
|
return -1
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func CudaMemCpyDtoH[T any](dst []T, src_d unsafe.Pointer, size int) int {
|
|
dst_c := unsafe.Pointer(&dst[0])
|
|
|
|
if err := C.cudaMemcpy(dst_c, src_d, C.size_t(size), 2); err != 0 {
|
|
return -1
|
|
}
|
|
return 0
|
|
}
|