Xuehai Pan
f3fce597e9
[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ ( #129769 )
...
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501 . Most changes are auto-generated by linter.
You can review these PRs via:
```bash
git diff --ignore-all-space --ignore-blank-lines HEAD~1
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129769
Approved by: https://github.com/ezyang
2024-08-04 10:24:09 +00:00
awayzjj
b70440f0a7
Document the torch.cuda.profiler.profile function ( #128216 )
...
Fixes https://github.com/pytorch/pytorch/issues/127901
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128216
Approved by: https://github.com/malfet , https://github.com/eqy
2024-06-17 23:42:40 +00:00
anandptl84
f48ca2561d
Document torch.cuda.profiler.start ( #128098 )
...
document https://github.com/pytorch/pytorch/issues/127917 start function of cuda/ profiler.py
Fixes 127917
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128098
Approved by: https://github.com/aaronenyeshi
2024-06-14 01:44:18 +00:00
anandptl84
0f52dc7e51
Document torch.cuda.profiler.stop ( #128196 )
...
Fixes #127918
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128196
Approved by: https://github.com/malfet , https://github.com/eqy
2024-06-12 17:39:43 +00:00
Aaron Orenstein
62bcdc0ac9
Flip default value for mypy disallow_untyped_defs [4/11] ( #127841 )
...
See #127836 for details.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/127841
Approved by: https://github.com/oulgen
2024-06-08 18:36:48 +00:00
Edward Z. Yang
3bf922a6ce
Apply UFMT to low traffic torch modules ( #106249 )
...
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106249
Approved by: https://github.com/Skylion007
2023-07-29 23:37:30 +00:00
Jianyu Huang
21f680e8ad
Follow up on CUDA 12 support for PyTorch/Caffe2 ( #95582 )
...
Differential Revision: D43610669
Pull Request resolved: https://github.com/pytorch/pytorch/pull/95582
Approved by: https://github.com/ngimel
2023-02-27 04:39:56 +00:00
anjali411
a6c0442cce
Add __all__ to torch.{autograd, fx, cuda} submodules ( #85343 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85343
Approved by: https://github.com/albanD
2022-10-09 14:46:54 +00:00
Alexander Grund
93719440b8
Replace map(lambda constructs ( #46462 )
...
Summary:
Follow-up of https://github.com/pytorch/pytorch/issues/46461 with a similar goal
Makes them more readable and possibly faster. Care has to be taken because `map` applies the function immediately while `(x for x in xs)` is a generator expression which gets evaluated later. This is a benefit in some cases where it is not required to actually create the list of values in memory (e.g. when passing to `tuple` or `extend` or `join`)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46462
Reviewed By: zou3519
Differential Revision: D24422343
Pulled By: ezyang
fbshipit-source-id: 252e33499c92ac0b15238f2df32681dbbda2b237
2020-10-22 09:50:22 -07:00
Nikita Shulga
8b5732e8ad
Move torch.cuda annotations inline ( #40075 )
...
Summary:
Also enable `torch.cuda` typechecking
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40075
Differential Revision: D22121275
Pulled By: malfet
fbshipit-source-id: dbecef09911334e8f3d87f5ecab66349da9f2325
2020-06-18 15:52:29 -07:00
Peter Bell
5fc5cf6571
Stop using ctypes to interface with CUDA libraries. ( #33678 )
...
Summary:
Fixes https://github.com/pytorch/pytorch/issues/33016 , Continuation of https://github.com/pytorch/pytorch/issues/31160
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33678
Differential Revision: D20249187
Pulled By: ezyang
fbshipit-source-id: 172ce4a0fee7fbe01436a421d1af22ef6173b6ed
2020-03-11 07:22:46 -07:00
Adam Paszke
76abc06b1f
Fix nvprof mode in autograd profiler
2017-10-20 10:22:54 -04:00
Adam Paszke
833bedc77d
Add CUDA profiler bindings
2017-09-25 23:21:30 -04:00
