File tree Expand file tree Collapse file tree 3 files changed +16
-13
lines changed Expand file tree Collapse file tree 3 files changed +16
-13
lines changed Original file line number Diff line number Diff line change 1
1
package cu
2
2
3
- // #include <cuda.h>
3
+ /*
4
+ #cgo CFLAGS: -Wno-deprecated-declarations
5
+
6
+ #include <cuda.h>
7
+ */
4
8
import "C"
5
9
import "unsafe"
6
10
Original file line number Diff line number Diff line change @@ -135,8 +135,8 @@ const (
135
135
// JITTarget11 JITTargetOption = C.CU_TARGET_COMPUTE_11
136
136
// JITTarget12 JITTargetOption = C.CU_TARGET_COMPUTE_12
137
137
// JITTarget13 JITTargetOption = C.CU_TARGET_COMPUTE_13
138
- JITTarget20 JITTargetOption = C .CU_TARGET_COMPUTE_20
139
- JITTarget21 JITTargetOption = C .CU_TARGET_COMPUTE_21
138
+ // JITTarget20 JITTargetOption = C.CU_TARGET_COMPUTE_20
139
+ // JITTarget21 JITTargetOption = C.CU_TARGET_COMPUTE_21
140
140
JITTarget30 JITTargetOption = C .CU_TARGET_COMPUTE_30
141
141
JITTarget32 JITTargetOption = C .CU_TARGET_COMPUTE_32
142
142
JITTarget35 JITTargetOption = C .CU_TARGET_COMPUTE_35
Original file line number Diff line number Diff line change @@ -3,6 +3,7 @@ package cu
3
3
/*
4
4
#include <cuda.h>
5
5
6
+ void handleCUDACB(void* fn);
6
7
void CallHostFunc(void* fn){
7
8
handleCUDACB(fn);
8
9
};
@@ -28,16 +29,14 @@ func (p *KernelNodeParams) c() *C.CUDA_KERNEL_NODE_PARAMS {
28
29
// here anonymous initialization of struct fields is used because `func` is a keyword.
29
30
// see also: https://github.com/golang/go/issues/41968
30
31
retVal := & C.CUDA_KERNEL_NODE_PARAMS {
31
- p .Func .fn ,
32
- C .uint (p .GridDimX ),
33
- C .uint (p .GridDimY ),
34
- C .uint (p .GridDimZ ),
35
- C .uint (p .BlockDimX ),
36
- C .uint (p .BlockDimY ),
37
- C .uint (p .BlockDimZ ),
38
- C .uint (p .SharedMemBytes ),
39
- nil ,
40
- nil ,
32
+ _func : p .Func .fn ,
33
+ gridDimX : C .uint (p .GridDimX ),
34
+ gridDimY : C .uint (p .GridDimY ),
35
+ gridDimZ : C .uint (p .GridDimZ ),
36
+ blockDimX : C .uint (p .BlockDimX ),
37
+ blockDimY : C .uint (p .BlockDimY ),
38
+ blockDimZ : C .uint (p .BlockDimZ ),
39
+ sharedMemBytes : C .uint (p .SharedMemBytes ),
41
40
}
42
41
return retVal
43
42
}
You can’t perform that action at this time.
0 commit comments