We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4a5d6cb commit 6be3f1fCopy full SHA for 6be3f1f
README.md
@@ -8,6 +8,23 @@ More Information / System Architecture:
8
9
- [GTC 2020 Talk](https://developer.nvidia.com/gtc/2020/video/s21671)
10
11
+## Example Usage
12
+
13
+```c++
14
+#include "torch/script.h"
15
+#include "trtorch/trtorch.h"
16
17
+...
18
+auto compile_settings = trtorch::ExtraInfo(dims);
19
+// FP16 execution
20
+compile_settings.op_precision = torch::kHalf;
21
+// Compile module
22
+auto trt_mod = trtorch::CompileGraph(ts_mod, compile_settings);
23
+// Run like normal
24
+auto results = trt_mod.forward({in_tensor});
25
26
+```
27
28
## Platform Support
29
30
| Platform | Support |
0 commit comments