pytorch/torch/utils/hipify
Jade Nie dcec099d48 Wrap Caffe2 (RowWise)SparseAdagrad fusion operator as a PT op (#39904)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/39904

This diff wraps Caffe2's (RowWise)SparseAdagrad fusion operator on GPU as a PT op.

Reviewed By: jianyuh

Differential Revision: D22010193

fbshipit-source-id: 5df3c506c0dadd3b21180829fd2d5084ac76abc3
2020-06-16 17:40:05 -07:00
..
__init__.py Move hipify to torch/utils to bundle them into torch package (#27425) 2019-10-07 17:25:45 -07:00
constants.py Move hipify to torch/utils to bundle them into torch package (#27425) 2019-10-07 17:25:45 -07:00
cuda_to_hip_mappings.py Wrap Caffe2 (RowWise)SparseAdagrad fusion operator as a PT op (#39904) 2020-06-16 17:40:05 -07:00
hipify_python.py Migrate pow kernel to c10::complex (#39286) 2020-06-11 15:49:30 -07:00