taco:Tensor代数编译器(taco)计算CPU和GPU上的稀疏张量表达式 源码
Tensor Algebra Compiler(taco)是一个C ++库,用于计算稀疏和密集张量上的张量代数表达式。 它使用新颖的编译器技术,与稀疏张量代数和稀疏线性代数的广泛使用的库中的手动优化内核在性能上具有竞争力。 您可以将taco用作C ++库,使您可以加载张量,从文件中读取张量以及计算张量表达式。 您也可以将taco用作代码生成器,该代码生成器生成用于计算张量表达式的C函数。 可以在,在论文中或在了解有关taco的更多信息。 要了解有关炸玉米饼近期发展趋势的更多信息,请参阅有关和的技术报告。 您还可以订阅电子邮件列表,我们在其中发布公告,RFC和API更改通知,或者订阅电子邮
文件列表
taco:Tensor代数编译器(taco)计算CPU和GPU上的稀疏张量表达式
(预估有个440文件)
transformations.cpp
39KB
intrinsic.cpp
40KB
index_notation.cpp
72KB
provenance_graph.cpp
49KB
merge_lattice.cpp
29KB
lowerer_impl.cpp
110KB
codegen_cuda.cpp
47KB
tensor.cpp
41KB
pyIndexNotation.cpp
34KB
setup.cfg
312B
暂无评论