Comments (11)
应该不会比tf慢吧。虽然没人会去单纯比较一个embedding层
from turbotransformers.
应该不会比tf慢吧。虽然没人会去单纯比较一个embedding层
请问大佬,使用GPU该如何设置?
if (device_type() == kDLCPU) {
.......
} else if (device_type() == kDLGPU) {
.......
}
from turbotransformers.
Python还是C++?python的话看example
from turbotransformers.
Python还是C++?python的话看example
例子是from_torch的,如果没有预训练模型,直接用tf Tensor转的core::Tensor,调用底层的算子呢?
from turbotransformers.
你如果已经把cpu model move to GPU了,然后再调用from_torch接口,那就自动在gpu上计算了。
from turbotransformers.
你如果实现把cpu model move to GPU了,然后再调用from_torch接口,那就自动在gpu上计算了。
我想单独测试一下turbo的embedding_lookup算子,自己实现了一个接口直接调用turbo的LookupEmbedding函数,但是每次都跑在CPU上,程序中没找到创建core::Tensor时设置device_type的接口
from turbotransformers.
首先你要让张量都放在cpu上,你可以调用Copy函数,显式转移数据。
其实真的duck不必单独加速embedding,就是加速到0延迟,也不会让你的bert变快。
See Amdahl's law
from turbotransformers.
首先你要让张量都放在cpu上,你可以调用Copy函数,显式转移数据。
其实真的duck不必单独加速embedding,就是加速到0延迟,也不会让你的bert变快。
See Amdahl's law
解决了。embedding_lookup确实不是热点,我想通过embedding算子的替换做一点技术积累,之后看能不能做成一个算子库,可以灵活选择的那种。未来希望和硬件、编译走的更近些
from turbotransformers.
TurboTransformers也需要人贡献算子库,比如conv1d这种算子还没实现呢,你愿不愿意involve进来?
from turbotransformers.
TurboTransformers也需要人贡献算子库,比如conv1d这种算子还没实现呢,你愿不愿意involve进来?
好啊,大佬给我个join方式吧
from turbotransformers.
readme底下有联系方式,你加我企业微信吧
from turbotransformers.
Related Issues (20)
- core::Tensor 处理 HOT 1
- 支持自己搭建的Transformer吗? HOT 1
- 请问怎么用tensorflow加载bert呢? HOT 1
- 多卡上会有问题?
- 支持大模型的推理吗 HOT 4
- Transformers版本不一致 HOT 5
- unitest失败 HOT 4
- Getting the same output logits regardless of input tensors HOT 14
- 想问下turbo支持huggingface的bart模型么 HOT 4
- RuntimeError: code is not compiled with CUDA. HOT 3
- MultiHeadedAttention与onmt结果不对应 HOT 1
- Is it easier to implement other model? conformer
- 如何在c++里使用profiler?
- 支持bert的变种吗? HOT 1
- Commercial Support? HOT 2
- Stable version HOT 4
- gpt2推理结果不正确 HOT 4
- C++的build问题 HOT 1
- arm cpu的支持情况。
- Support for ViT and Swin transformer ?
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from turbotransformers.