TensorComprehensions:一种表示机器学习工作负载的领域特定语言 源码

Tensor Comprehensions(TC)是一个功能齐全的C ++库,可使用 , 和NVRTC或LLVM自动合成高性能的机器学习内核。 TC还提供了与Caffe2和PyTorch的基本集成。 我们在关于的论文中提供了更多细节。 该库被设计为高度可移植的,与机器学习框架无关的,并且仅需要具有内存分配,卸载和同步功能的简单张量库。 目前,我们已经将TC与和集成。 一个简单的例子 以下内容说明了该库的一个简短而强大的功能:针对特定大小,可以按需JIT编译高性能机器学习内核的能力。 import tensor_comprehensions as tc import torch lang