tensorrt-einsumplugin's People
Forkers
wuqiangch ko440124 awfsgdf zhaohb kangolhsu ltyanghuang vuminhduc97 blue-q hqu-cst-mmc tream733tensorrt-einsumplugin's Issues
关于时空图卷积神经网络中的einsum运算等式
你好,在时空图卷积神经网络的操作应该是“nkctv,kvw->nctw”,而你的代码中equation == "nctkv,kvw->nctw"。
Inference Speed
Hello, thank for customing Einsum layer!
Is Inference Speed good? I used to change this layer by mul & sum, But Inference Speed significantly reduced.
Cuda Error in findFastestTactic: 700
Have you met below error? I add Einsum Plugin to TensorRT Lib successfully. But when I convert and serialize onnx--> engine, I met this error:
[E] [TRT] ../builder/cudnnBuilderUtils.cpp (426) - Cuda Error in findFastestTactic: 700 (an illegal memory access was encountered)
[E] [TRT] ../rtSafe/safeRuntime.cpp (32) - Cuda Error in free: 700 (an illegal memory access was encountered)
terminate called after throwing an instance of 'nvinfer1::CudaError'
在刚刚发布的TensorRT 8.2 EA版本已经支持Einsum算子的大部分等式操作
在刚刚发布的TensorRT 8.2 EA版本已经支持Einsum算子的大部分等式操作
如果queue里面要调用cu函数,应该要做些什么?
请问如果要在queue里面调用cu函数的话,那个cmakelists文件应该怎么写呀?
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.