Skip to content

Commit d2f747c

Browse files
committed
add TBLIS docstring
1 parent 118e5d3 commit d2f747c

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/TensorOperationsTBLIS.jl

+8-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,14 @@ set_num_tblis_threads(n) = LibTBLIS.tblis_set_num_threads(convert(Cuint, n))
2323

2424
# TensorOperations
2525
#------------------
26-
26+
"""
27+
TBLIS()
28+
29+
Backend for tensor operations on strided arrays (arrays that can be cast into a `StridedView`)
30+
that uses [TBLIS](https://github.com/devinamatthews/tblis) library. This library can perform
31+
tensor contractions without additional intermediate tensors. It does not use LinearAlgebra
32+
or BLAS.
33+
"""
2734
struct TBLIS <: TensorOperations.AbstractBackend end
2835

2936
Base.@deprecate(tblisBackend(), TBLIS())

0 commit comments

Comments
 (0)