acrotensor:一个用于计算大规模张量收缩的C ++库

时间:2024-05-30 11:14:02
【文件属性】:

文件名称:acrotensor:一个用于计算大规模张量收缩的C ++库

文件大小:129KB

文件格式:ZIP

更新时间:2024-05-30 11:14:02

C++

肩cro肌 Acrotensor是一个C ++ / CUDA库,用于计算CPU和GPU上的任意张量收缩。 张量是动态调整大小的,以提供高度的灵活性,并且使用自然的数学符号定义张量收缩以实现最大的可用性。 为了保持良好的性能,收缩代码是动态生成的,具有固定的大小和展开的循环,然后编译即时(JIT)以产生更好的优化执行。 入门 Acrotensor依赖于C ++ 11编译器,并且需要在编译器上使用nvcc CUDA包装器才能处理C ++和CUDA的混合体。 要开始构建,您需要输入acrotensor目录并运行: make config 这将生成一个config/config.mk文件,其中包含针对您的环境可能需要更改的一组默认值。 编辑config.mk只需输入acrotensor目录并运行: make 这将建立可链接到lib文件夹中的静态和动态库,并生成包含所有需要的头文件的inc目录。


【文件预览】:
acrotensor-master
----exec()
--------KernelExecutor.hpp(2KB)
--------CudaExecutor.hpp(2KB)
--------KernelExecutor.cpp(3KB)
--------CPUInterpretedExecutor.cpp(13KB)
--------CudaExecutor.cpp(23KB)
--------CPUInterpretedExecutor.hpp(2KB)
--------Executor.hpp(392B)
----kernel()
--------DimensionedMultiKernel.hpp(5KB)
--------DimensionedKernel.cpp(7KB)
--------TensorEngine.cpp(24KB)
--------TensorKernel.cpp(10KB)
--------TensorKernel.hpp(4KB)
--------DimensionedKernel.hpp(3KB)
--------TensorEngine.hpp(7KB)
--------DimensionedMultiKernel.cpp(10KB)
----LICENSE(1KB)
----util()
--------Util.hpp(430B)
--------StringUtil.hpp(898B)
--------CudaUtil.hpp(5KB)
--------CudaUtil.cpp(6KB)
--------Error.hpp(913B)
----README.md(4KB)
----ops()
--------NativeCPUOps.cpp(3KB)
--------NativeCPUOps.hpp(2KB)
--------Ops.hpp(350B)
--------CudaGPUOps.cpp(8KB)
--------NonContractionOps.hpp(1KB)
--------CudaGPUOps.hpp(2KB)
----AcroTensor.hpp(394B)
----config()
--------defaults.mk(710B)
----tensor()
--------Tensor.hpp(11KB)
--------IndexMapping.hpp(2KB)
--------IndexMapping.cpp(2KB)
--------Tensor.cpp(10KB)
--------IndexVector.hpp(2KB)
--------IndexVector.cpp(3KB)
--------SliceTensor.hpp(2KB)
--------SliceTensor.cpp(5KB)
----makefile(2KB)
----unittest()
--------catch.hpp(313KB)
--------kernel()
--------LICENSE_1_0.txt(30KB)
--------unit_test_main.cpp(419B)
--------tensor()
--------makefile(1KB)

网友评论