OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
f77783c374
pytorch/torch/utils
History
Andrey Talman 46f9e16afe Documenting cuda 11.5 windows issue (#73013) 
Summary:
Adding documentation about compiling extension with CUDA 11.5 and Windows

Example of failure: https://github.com/pytorch/pytorch/runs/4408796098?check_suite_focus=true

 Note: Don't use torch/extension.h In CUDA 11.5 under windows in your C++ code:
    Use aten instead of torch interface in all cuda 11.5 code under windows. It has been failing with errors, due to a bug in nvcc.
    Example use:
        >>> #include <ATen/ATen.h>
        >>> at::Tensor SigmoidAlphaBlendForwardCuda(....)
    Instead of:
        >>> #include <torch/extension.h>
        >>> torch::Tensor SigmoidAlphaBlendForwardCuda(...)
    Currently open issue for nvcc bug: https://github.com/pytorch/pytorch/issues/69460
    Complete Workaround code example: cb170ac024

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73013

Reviewed By: malfet, seemethere

Differential Revision: D34306134

Pulled By: atalman

fbshipit-source-id: 3c5b9d7a89c91bd1920dc63dbd356e45dc48a8bd
(cherry picked from commit 87098e7f17)
2022-02-19 02:34:59 +00:00
..
backcompat
benchmark Convert type comments to annotations in caffe2/torch/util (#72667) 2022-02-11 20:50:20 +00:00
bottleneck [profiler][refactor] Refactor the usage of legacy profiler implementation (#61931) 2021-08-03 18:51:29 -07:00
data [DataPipe] Adding usage examples for IterDataPipes (#73033) 2022-02-18 15:12:34 +00:00
ffi
hipify [PyTorch] Support NVTX range_start and range_end (#70030) 2022-02-07 17:31:57 +00:00
model_dump fix model dump for the lowered module (#72866) 2022-02-17 07:09:44 +00:00
tensorboard Stop writing logs to root logger (#72649) 2022-02-11 21:30:53 +00:00
__init__.py Fix breakpad build and add to more images (#59236) 2021-06-01 22:47:14 -07:00
_cpp_extension_versioner.py
_crash_handler.py Use CMake for breakpad (#63186) 2021-08-19 10:42:01 -07:00
_freeze.py don't leak build time path name to runtime for frozen python modules (#65715) 2021-09-28 19:25:51 -07:00
_python_dispatch.py [Reland] Add python mode (#64360) 2021-09-16 09:02:30 -07:00
_pytree.py Teach pytrees about namedtuple (#62292) 2021-07-28 06:27:44 -07:00
_zip.py Load zip file in deploy interpreter (#71072) 2022-01-15 14:39:59 -08:00
bundled_inputs.py [SmartCompose][OnDevice]fix function name bug in mobile export & Script to convert mobile model (#66915) 2021-10-20 18:14:51 -07:00
checkpoint.py Propagate full autocast state to CheckpointFunction's forward-inside-backward (#71169) 2022-01-27 00:31:53 +00:00
collect_env.py Improve pip package determination (#63321) 2021-08-16 13:54:39 -07:00
cpp_extension.py Documenting cuda 11.5 windows issue (#73013) 2022-02-19 02:34:59 +00:00
dlpack.py Improve docs for from_dlpack and to_dlpack (#70437) 2022-01-25 20:32:12 +00:00
file_baton.py
hooks.py Fix full backward hook when grad is disabled (#65335) 2021-09-20 13:31:19 -07:00
mkldnn.py
mobile_optimizer.py Small fix type hints in mobile optimizer (#59282) 2021-06-02 15:32:16 -07:00
model_zoo.py
show_pickle.py show_pickle/model_dump: Handle invalid UTF-8 in pickles (#57661) 2021-06-04 19:42:25 -07:00
throughput_benchmark.py