pytorch/torch/utils/hipify
Jeff Daily 7c7b2d89ba [ROCm] set hipblas workspace (#138791)
Fixes #138532.

This brings hipblas behavior in line with cublas behavior with respect to setting the workspace to an allocation from the caching allocator as well as the env var HIPBLAS_WORKSPACE_CONFIG.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/138791
Approved by: https://github.com/naromero77amd, https://github.com/eqy, https://github.com/malfet

Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
2024-10-29 01:37:55 +00:00
..
__init__.py
constants.py Revert "[ROCm] remove caffe2 from hipify (#137157)" 2024-10-08 17:45:45 +00:00
cuda_to_hip_mappings.py [ROCm] set hipblas workspace (#138791) 2024-10-29 01:37:55 +00:00
hipify_python.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
version.py