Home
last modified time | relevance | path

Searched defs:OptionalHIPGuardMasqueradingAsCUDA (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h271 struct OptionalHIPGuardMasqueradingAsCUDA { struct
272 explicit OptionalHIPGuardMasqueradingAsCUDA() : guard_() {} in OptionalHIPGuardMasqueradingAsCUDA() argument
273 …explicit OptionalHIPGuardMasqueradingAsCUDA(std::optional<Device> device_opt) : guard_(device_opt)… in OptionalHIPGuardMasqueradingAsCUDA() argument
274 …explicit OptionalHIPGuardMasqueradingAsCUDA(std::optional<DeviceIndex> device_index_opt) : guard_(… in OptionalHIPGuardMasqueradingAsCUDA() argument
277 OptionalHIPGuardMasqueradingAsCUDA& operator=(const OptionalHIPGuardMasqueradingAsCUDA&) = delete; argument
279OptionalHIPGuardMasqueradingAsCUDA& operator=(OptionalHIPGuardMasqueradingAsCUDA&& other) = delete; argument
281 void set_device(Device device) { guard_.set_device(device); } in set_device()
282 void reset_device(Device device) { guard_.reset_device(device); } in reset_device()
283 void set_index(DeviceIndex device_index) { guard_.set_index(device_index); } in set_index()
284 std::optional<Device> original_device() const { return guard_.original_device(); } in original_device()
[all …]