Comments (3)
I was thinking we might just put set_num_threads
and get_num_threads
in TH instead of compiling the python extension with OpenMP. Thoughts?
from pytorch.
yea that's fine with me. these things are better controlled with OMP_NUM_THREADS / MKL_NUM_THREADS anyways, as then the blas libraries will also limit their cores.
from pytorch.
moving to TH seems much better to me than compiling with OMP.
from pytorch.
Related Issues (20)
- Typinghints on tensor arithmetic operations (other:Any) cause incompatibilities
- Expand stale management to Issues HOT 2
- FSDP RuntimeError Output 0 of ViewBackward0 is a view and its base or another view of its base has been modified inplace. HOT 5
- Fix OOMs during parallel builds
- [FSDP+TP] RuntimeError: 'weight' must be 2-D HOT 1
- Inconsistency error of `torch.Tensor.atan2_` with torch.compile HOT 1
- CUDA kernel failed: no kernel image is available for execution on the device HOT 1
- Different results for single sample vs. batched HOT 3
- rand_like on permuted parameter not working on MPS
- running AOTI-generated DSO causes segmentation fault on Linux x86, passes on ARM MacOS HOT 6
- can not find the caffe2::Threads
- torch.compile error HOT 2
- dlopen: cannot load any more object with static TLS
- Compilation Error in release/1.7 due to Missing Definition for OneCycleLR in lr_scheduler.pyi HOT 3
- MPS slows down after sleep HOT 1
- MPS memory leak?
- [RFC] Support Torchrun for Out-of-the-tree Devices
- DISABLED test_batched_mm_bfloat16_bs_4_cuda_bfloat16 (__main__.TestDecompCUDA) HOT 2
- DISABLED test_expand_cuda (__main__.TestUnbackedSymintsCUDA) HOT 1
- DISABLED test_binary_op_list_error_cases__foreach_add_cuda (__main__.TestForeachCUDA) 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.