Comments (2)
Now located in https://github.com/csarofeen/pytorch/blob/20_6_11_devel/torch/csrc/jit/codegen/cuda/lower_unroll.cpp#L27
Maybe @tlemo can help out with this as he reviews lowering.
from pytorch.
With the big indexing rework this code should be relatively clear and well commented.
from pytorch.
Related Issues (20)
- `full` python API is broken and doesn't print a proper definition HOT 1
- Segfault when using addcmul in the frontend
- disable welford translate if the original scheduler before translate is not persistent
- nearbyint fails with NVRTC compile error for integer inputs
- cache var used by each iteration in grid persistent kernel, e.g. weight in layer norm backward
- warp reduction for complex numbers
- Welford for complex numbers HOT 1
- Improve shared memory reuse
- `replay_swizzle` argument should come after `error_on_failure`
- broadcast_in_dim: The size of contiguity must equal to the number of non-broadcasting IterDomains HOT 6
- Failed test cases on H100, node cl1-2547
- ValReplacementMutator fails to mutate an expression even when its output is mutated HOT 1
- Example of Gather/Scatter from CrossEntropyLoss HOT 14
- Take internal buffer size into account to decide on indexing type size
- codegen error: RuntimeError: producer->getMemoryType() == MemoryType::Global || producer->getMemoryType() == MemoryType::Shared INTERNAL ASSERT FAILED HOT 3
- Indexing failure HOT 1
- Compilation failure with real()/imag() on complex64
- detect parallel pattern error and provide a more helpful message HOT 1
- Get Segmentation fault when I compile the following FusionDefinition HOT 2
- Update nvFuser number math to be in float
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.