projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 02a249e) | patch
Added deviceCount() virtual method to DeviceGuardImplInterface (#15574)
authorIgor Fedan <ifedan@fb.com>
Thu, 27 Dec 2018 23:24:22 +0000 (15:24 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 27 Dec 2018 23:36:32 +0000 (15:36 -0800)
commit62151aa259154a9206d812b16525fd5b55ce4a89
tree06e20915bef1b664debad8908403df0e72078d81tree | snapshot
parent02a249ed922018a22c77ee8b106205027de93cc4commit | diff
Added deviceCount() virtual method to DeviceGuardImplInterface (#15574)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15574

Added deviceCount() virtual method to DeviceGuardImplInterface, also added correspondent implementation for CPUGuardImpl, CUDAGuardImpl, FakeGuardImpl, VirtualGuardImpl, HIPGuardImplMasqueradingAsCUDA

Reviewed By: soumith

Differential Revision: D13554609

fbshipit-source-id: 913bf2aad44a0a356efe54505ee4abaf6c4622db
aten/src/ATen/detail/CPUGuardImpl.h diff | blob | history
aten/src/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.h diff | blob | history
c10/cuda/impl/CUDAGuardImpl.h diff | blob | history
c10/impl/DeviceGuardImplInterface.h diff | blob | history
c10/impl/FakeGuardImpl.h diff | blob | history
c10/impl/VirtualGuardImpl.h diff | blob | history
Domain: Machine Learning / ML Framework;