Skip to content

Commit 17806da

Browse files
committed
Update cutensor.toml for block sparse contraction
1 parent 3c11bec commit 17806da

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

res/wrap/cutensor.toml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,3 +57,10 @@ needs_context = false
5757
6 = "CuPtr{Cvoid}"
5858
7 = "CuPtr{Cvoid}"
5959
8 = "CuPtr{Cvoid}"
60+
61+
[api.cutensorBlockSparseContract.argtypes]
62+
4 = "CuPtr{Cvoid}"
63+
5 = "CuPtr{Cvoid}"
64+
7 = "CuPtr{Cvoid}"
65+
8 = "CuPtr{Cvoid}"
66+
9 = "CuPtr{Cvoid}"

0 commit comments

Comments
 (0)