Comments (3)
cc @ezyang if it makes sense to do not perform 0/1 specialization on nn module attributes, and directly make any int attribute SymInt.
@yanboliang because I remember you sent a PR for this.
from pytorch.
Turning this into high priority because I expect this is a common pattern and could surprise users.
from pytorch.
It's relatively simple to get this in one go, but we are changing a heuristic. I opened a PR to check the blast radius. I'll add your test in if we decide we can land it.
from pytorch.
Related Issues (20)
- [inductor] assert_close numerics failed when fusing abs,max,clamp,mul,reciprocal (needed by float8) HOT 7
- "bmm_sparse_cuda" not implemented for 'BFloat16'
- Unable to Specify CUDA Stream for Collective Operations Using with torch.cuda.stream() context HOT 4
- linux-aarch64 CI tests are being timed out resulting in test failures HOT 6
- [Bug][CompiledAutograd] PyTorch with CompiledAutograd throws an IndexError: list index out of range when the hook code includes a list append operation. HOT 1
- Compiling flex attention with batch dependent block mask and dynamic shapes
- Compilation with USE_MPI throws CMake errors HOT 1
- ProcessPoolExecutor resets all Tensors on the cuda device when using a cuda tensor HOT 1
- Compile with USE_CPP_CODE_COVERAGE=ON throw erros HOT 2
- [Inductor][FP8] AOTInductor appears to ignore a transpose in `test_fp8_view_of_param_non_abi_compatible_cuda` HOT 1
- [torch.export] Automate export constrains like in onnx dynamo HOT 1
- triangular ops cuda kernel int overflow HOT 1
- [torch.export] Detect internal constrains HOT 6
- Segmentaion fault on Apple M1 caused by libomp on v2.4.1 HOT 6
- [pipelining] create two stage instances out of same stage module HOT 1
- [pipelining] try not to dry run module when creating PipelineStage? HOT 3
- [Inductor] Some mechanism to fuse pointwise ops (and more) into a user-defined triton kernel
- Can not load full state dictionary containing parameters whose first dimension is not divisible by DP size in FSDP 2 + TP model HOT 2
- DISABLED test_mem_leak_from_save_for_bw (__main__.TestAOTAutogradWithDynamo) HOT 1
- DISABLED test_graph_grad_scaling_foreach_False_fused_True_AdamW_cuda_float32 (__main__.TestCudaOptimsCUDA) HOT 2
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.