Skip to content

Commit e7bbc2d

Browse files
committed
chore: More README updates
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent ded366d commit e7bbc2d

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

core/plugins/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@ auto interpolate_plugin = creator->createPlugin(name, &fc); // fc is the collect
3131
If you'd like to compile your plugin with Torch-TensorRT,
3232

3333
- Add your implementation to the `impl` directory
34-
- Add a call `REGISTER_TRTORCH_PLUGINS(MyPluginCreator)` to `register_plugins.cpp`. `MyPluginCreator` is the plugin creator class which creates your plugin. By adding this to `register_plugins.cpp`, your plugin will be initialized and accessible (added to TensorRT plugin registry) during the `libtorchtrt_plugins.so` library loading.
34+
- Add a call `REGISTER_TORCHTRT_PLUGIN(MyPluginCreator)` to `register_plugins.cpp`. `MyPluginCreator` is the plugin creator class which creates your plugin. By adding this to `register_plugins.cpp`, your plugin will be initialized and accessible (added to TensorRT plugin registry) during the `libtorchtrt_plugins.so` library loading.
3535
- Update the `BUILD` file with the your plugin files and dependencies.
3636
- Implement a converter op which makes use of your plugin.
3737

3838
Once you've completed the above steps, upon successful compilation of Torch-TensorRT library, your plugin should be available in `libtorchtrt_plugins.so`.
3939

40-
A sample runtime application on how to run a network with plugins can be found <a href="https://github.com/NVIDIA/Torch-TensorRT/tree/master/examples/trtorchrt_example" >here</a>
40+
A sample runtime application on how to run a network with plugins can be found <a href="https://github.com/NVIDIA/Torch-TensorRT/tree/master/examples/torchtrt_runtime_example" >here</a>

cpp/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,5 +18,5 @@ bazel build //cpp/lib:libtorchtrt.so -c opt
1818
## Usage
1919

2020
``` c++
21-
#include "trtorch/trtorch.h"
21+
#include "torch_tensorrt/torch_tensorrt.h"
2222
```

cpp/bin/torchtrtc/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ output two formats, either a TorchScript program with the TensorRT engine embedd
88
the TensorRT engine itself as a PLAN file.
99

1010
All that is required to run the program after compilation is for C++ linking against libtorchtrt.so
11-
or in Python importing the trtorch package. All other aspects of using compiled modules are identical
11+
or in Python importing the torch_tensorrt package. All other aspects of using compiled modules are identical
1212
to standard TorchScript. Load with `torch.jit.load()` and run like you would run any other module.
1313

1414

0 commit comments

Comments
 (0)