5 Commits

Author	SHA1	Message	Date
rohithkrn	c528faac7d	[ROCm] Skip problematic mgpu tests on ROCm3.5 (#41409) ... Summary: nccl tests and parallelize_bmuf_distributed test are failing on rocm3.5.1. Skipping these tests to upgrade the CI to rocm3.5.1 jeffdaily sunway513 Pull Request resolved: https://github.com/pytorch/pytorch/pull/41409 Reviewed By: orionr Differential Revision: D22528928 Pulled By: seemethere fbshipit-source-id: 928196b7a62a441d391e69f54b278313ecc75d77	2020-07-14 11:55:43 -07:00
rohithkrn	aa88c2c0b6	Unify gpu_support variable in python tests (#16748) ... Summary: Assign `has_gpu_support = has_cuda_support or has_hip_support` and make according changes in python tests. Pull Request resolved: https://github.com/pytorch/pytorch/pull/16748 Differential Revision: D13983132 Pulled By: bddppq fbshipit-source-id: ca496fd8c6ae3549b736bebd3ace7fa20a6dad7f	2019-02-07 00:29:51 -08:00
rohithkrn	0d663cec30	Unify cuda and hip device types in Caffe2 python front end (#14221) ... Summary: Goal of this PR is to unify cuda and hip device types in caffe2 python front end. Pull Request resolved: https://github.com/pytorch/pytorch/pull/14221 Differential Revision: D13148564 Pulled By: bddppq fbshipit-source-id: ef9bd2c7d238200165f217097ac5727e686d887b	2018-11-29 14:00:16 -08:00
Orion Reblitz-Richardson	1d5780d42c	Remove Apache headers from source. ... * LICENSE file contains details, so removing from individual source files.	2018-03-27 13:10:18 -07:00
Qinqing Zheng	4471e15b76	BMUF cpu support ... Summary: change the interface so BMUF can run on cpus Reviewed By: asaadaldien Differential Revision: D6356026 fbshipit-source-id: f58a4da9f800d969145a1a376e118b0f3581f8c1	2017-11-19 23:41:25 -08:00