TBLIS.jl:用于TBLIS张量收缩库的Julia包装器

时间:2024-03-17 22:36:34
【文件属性】:

文件名称:TBLIS.jl:用于TBLIS张量收缩库的Julia包装器

文件大小:2.49MB

文件格式:ZIP

更新时间:2024-03-17 22:36:34

Julia

TBLIS.jl 用于TBLIS张量收缩库的Julia包装器。 [!]进行中在Julia夜间版本和最新稳定版本(1.4.2)上经过测试 目前仅实现张量加法和乘法-如果您需要TBLIS的其他功能,请提出问题。 安装 julia>] pkg>add TBLIS 用法 using TBLIS TBLIS.init() TT = Float32 O = 10 V = 50 _A = rand(TT,O,O,O,O) _B = rand(TT,O,O,V,V) _C = zeros(TT,O,O,V,V) 数组必须转换为TTensor对象。 A = TBLIS.TTensor{TT}(_A) B = TBLIS.TTensor{TT}(_B) C = TBLIS.TTensor{TT}(_C) 该对象仅创建一个指向原始数据的指针。 julia> A.data === _A true 对于一般


【文件预览】:
TBLIS.jl-master
----.github()
--------workflows()
----src()
--------libtblis.dylib(8.47MB)
--------TBLIS.jl(833B)
--------TTensor.jl(2KB)
--------libtci.dylib(21KB)
--------Ops.jl(867B)
----Project.toml(444B)
----.travis.yml(530B)
----test()
--------runtests.jl(20B)
--------test1.jl(998B)
----Manifest.toml(2KB)
----README.md(1KB)

网友评论