Comments (2)
You can simplify the usage with a contextmanager
@contextmanager
def dump_cudagraph(graph, output_path):
try:
graph.enable_debug_dump()
yield
finally:
graph.debug_dump(output_path)
g = torch.cuda.CUDAGraph()
with dump_cudagraph(g, "out.txt"), torch.cuda.graph(g):
from pytorch.
Yup, I can do it. Just wonder why pytorch does not have such a builtin support. It's pretty simple to implement.
The current cudagraph API in pytorch, especially with respect to debug dump, is not straightforward.
from pytorch.
Related Issues (20)
- Internal uses of `torch.load` are missing `weights_only` and raise FutureWarning HOT 4
- Internal uses of `torch.cuda.amp.autocast` raise FutureWarnings HOT 9
- `atleast_1d()` with no arguments works returning an empty tuple against the doc HOT 1
- `atleast_2d()` with no arguments works returning an empty tuple against the doc HOT 1
- `atleast_{1,2,3}d()` with no arguments works returning an empty tuple against the doc HOT 1
- output=model(input) report error:'gbk' codec can't decode byte 0x8a in position 171: illegal multibyte sequence HOT 1
- DTensor does not support bit shift HOT 1
- Torch C++ extension build failed with `fatal error: nlohmann/json.hpp: No such file or directory`
- TorchScript: `Return value was annotated as having type float but is actually of type int` violates PEP 484
- Pytorch 2.4 RC cu118 wheels do not work on old drivers HOT 3
- The doc of `stack()` should say `tuple` or `list` of tensors for `tensors` argument HOT 1
- Compilation Fails with torch.sparse and "fullgraph=True" HOT 4
- The doc of `cat()` should say `tuple` or `list` of tensors for `tensors` argument HOT 1
- The doc of `hstack()` should say `tuple` or `list` of tensors for `tensors` argument HOT 1
- The doc of `vstack()` should say `tuple` or `list` of tensors for `tensors` argument HOT 1
- The doc of `column_stack()` should say `tuple` or `list` of tensors for `tensors` argument HOT 1
- The doc of `dstack()` should say `tuple` or `list` of tensors for `tensors` argument HOT 1
- fx.wrap() doesn't really work for things in torch/*
- Converting a numpy array of size larger than 32,768 to a tensor causes a segmentation fault HOT 7
- tree_map_only_ doesn't seem work as expected HOT 5
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 pytorch.