You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Currently provides implementations of `tensorcontract!`, `tensoradd!` and `tensortrace!` for
17
-
`StridedArray{<:BlasFloat}`. These can be accessed through the backend system of
18
-
TensorOperations, i.e.
16
+
Currently provides implementations of `tensorcontract!`, `tensoradd!` and `tensortrace!` for array types compatible with Strided.jl, i.e. `StridedView{<:BlasFloat}`.
17
+
These can be accessed through the backend system of TensorOperations, i.e.
19
18
20
19
```julia
21
20
using TensorOperations
22
21
using TensorOperationsTBLIS
23
22
24
-
tblisbackend =tblisBackend()
23
+
tblisbackend =TBLIS()
25
24
α =randn()
26
25
A =randn(5, 5, 5, 5, 5, 5)
27
26
B =randn(5, 5, 5)
@@ -34,17 +33,16 @@ D = zeros(5, 5, 5)
34
33
end
35
34
```
36
35
37
-
Additionally, the number of threads used by tblis can be set by:
36
+
Additionally, the number of threads used by TBLIS can be set by:
38
37
39
38
```julia
40
-
using TensorOperationsTBLIS
41
-
tblis_set_num_threads(4)
42
-
@showtblis_get_num_threads()
39
+
TensorOperationsTBLIS.set_num_threads(4)
40
+
@show TensorOperationsTBLIS.get_num_threads()
43
41
```
44
42
45
43
## Notes
46
44
47
-
- This implementation of tblis for TensorOperations.jl is only supported from v5 of
45
+
- This implementation of TBLIS for TensorOperations.jl is only supported from v5 of
48
46
TensorOperations.jl onwards. For v4, an earlier version of this package exists.
49
47
For older versions, you could look for
50
48
[BliContractor.jl](https://github.com/xrq-phys/BliContractor.jl) or
0 commit comments