mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
37c6087334
## Summary - add a CuBLASReductionOption enum so the CUDA context can track reduced-precision and split-K options - extend the Python bindings, backend helpers, and docs to accept an optional allow_splitk argument for fp16/bf16 matmul controls - update cuBLAS/cuBLASLt call sites plus dynamo guards and tests to respect the new combinations ## Testing - python test/test_cuda.py TestCuda.test_cublas_allow_fp16_reduced_precision_reduction_get_set -v *(fails: ModuleNotFoundError: No module named 'psutil')* ------ https://chatgpt.com/codex/tasks/task_e_68e404623178832f8a3e1d34e1e175da Pull Request resolved: https://github.com/pytorch/pytorch/pull/164766 Approved by: https://github.com/malfet, https://github.com/albanD |
||
|---|---|---|
| .. | ||
| cache_entry.cpp | ||
| cache_entry.h | ||
| compiled_autograd.cpp | ||
| compiled_autograd.h | ||
| cpp_shim.cpp | ||
| cpp_shim.h | ||
| cpython_defs.c | ||
| cpython_defs.h | ||
| cpython_includes.h | ||
| debug_macros.h | ||
| eval_frame_cpp.cpp | ||
| eval_frame_cpp.h | ||
| eval_frame.c | ||
| eval_frame.h | ||
| extra_state.cpp | ||
| extra_state.h | ||
| framelocals_mapping.cpp | ||
| framelocals_mapping.h | ||
| guards.cpp | ||
| guards.h | ||
| init.cpp | ||
| init.h | ||
| python_compiled_autograd.cpp | ||
| python_compiled_autograd.h | ||
| utils.cpp | ||
| utils.h | ||