-
Notifications
You must be signed in to change notification settings - Fork 1
/
peer.go
51 lines (42 loc) · 1.3 KB
/
peer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package cu
// This file implements CUDA unified addressing.
//#include <cuda.h>
import "C"
import (
"unsafe"
)
// Make allocations from the peer Context available to the current context.
func CtxEnablePeerAccess(peer Context) {
err := Result(C.cuCtxEnablePeerAccess(C.CUcontext(unsafe.Pointer(uintptr(peer))), C.uint(0)))
if err != SUCCESS {
panic(err)
}
}
// Make allocations from the peer Context available to the current context.
func (peer Context) EnablePeerAccess() {
CtxEnablePeerAccess(peer)
}
// Reverses CtxEnablePeerAccess().
func CtxDisablePeerAccess(peer Context) {
err := Result(C.cuCtxDisablePeerAccess(C.CUcontext(unsafe.Pointer(uintptr(peer)))))
if err != SUCCESS {
panic(err)
}
}
// Reverses EnablePeerAccess().
func (peer Context) DisablePeerAccess() {
CtxDisablePeerAccess(peer)
}
// Returns true if CtxEnablePeerAccess can be called on a context for dev and peerDev.
func DeviceCanAccessPeer(dev, peer Device) bool {
var canAccessPeer C.int
err := Result(C.cuDeviceCanAccessPeer(&canAccessPeer, C.CUdevice(dev), C.CUdevice(peer)))
if err != SUCCESS {
panic(err)
}
return int(canAccessPeer) != 0
}
// Returns true if CtxEnablePeerAccess can be called on a context for dev and peerDev.
func (dev Device) CanAccessPeer(peer Device) bool {
return DeviceCanAccessPeer(dev, peer)
}