GitLab

* modified code for using ROCm backened within the PyTorch framework

* added hip runtime header

* flake8 linting fix