16 lines
300 B
Go
16 lines
300 B
Go
// NOTE: functions in this file would be automatically generated
|
|
// and named as `c-generated.go`
|
|
package libtch
|
|
|
|
//#include "stdbool.h"
|
|
//#include "torch_api.h"
|
|
import "C"
|
|
import (
|
|
"unsafe"
|
|
)
|
|
|
|
func Atc_cuda_device_count() int {
|
|
t := C.atc_cuda_device_count()
|
|
return *(*int)(unsafe.Pointer(&t))
|
|
}
|