CUDA.jl/res/wrap/cutensor.toml at 3316f636928991ec005f69cbb475f9445a9e113b · JuliaGPU/CUDA.jl · GitHub

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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
[general]
library_name = "libcutensor"
output_file_path = "../../lib/cutensor/src/libcutensor.jl"
prologue_file_path = "./libcutensor_prologue.jl"
epilogue_file_path = "./libcutensor_epilogue.jl"

output_ignorelist = [
    # these change often
    "CUTENSOR_VERSION",
    "CUTENSOR_MAJOR",
    "CUTENSOR_MINOR",
    "CUTENSOR_PATCH",
]


[codegen]
use_ccall_macro = true
always_NUL_terminated_string = true


[api]
checked_rettypes = [ "cutensorStatus_t" ]

[api.cutensorGetErrorString]
needs_context = false

[api.cutensorGetVersion]
needs_context = false

[api.cutensorGetCudartVersion]
needs_context = false

[api.cutensorElementwiseTrinaryExecute.argtypes]
4 = "CuPtr{Cvoid}"
6 = "CuPtr{Cvoid}"
8 = "CuPtr{Cvoid}"
9 = "CuPtr{Cvoid}"

[api.cutensorElementwiseBinaryExecute.argtypes]
4 = "CuPtr{Cvoid}"
6 = "CuPtr{Cvoid}"
7 = "CuPtr{Cvoid}"

[api.cutensorPermute.argtypes]
4 = "CuPtr{Cvoid}"
5 = "CuPtr{Cvoid}"

[api.cutensorContract.argtypes]
4 = "CuPtr{Cvoid}"
5 = "CuPtr{Cvoid}"
7 = "CuPtr{Cvoid}"
8 = "CuPtr{Cvoid}"
9 = "CuPtr{Cvoid}"

[api.cutensorReduce.argtypes]
4 = "CuPtr{Cvoid}"
6 = "CuPtr{Cvoid}"
7 = "CuPtr{Cvoid}"
8 = "CuPtr{Cvoid}"

[api.cutensorBlockSparseContract.argtypes]
4 = "Ptr{CuPtr{Cvoid}}"
5 = "Ptr{CuPtr{Cvoid}}"
7 = "Ptr{CuPtr{Cvoid}}"
8 = "Ptr{CuPtr{Cvoid}}"
9 = "Ptr{CuPtr{Cvoid}}"