Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h316 struct OptionalHIPStreamGuardMasqueradingAsCUDA { struct
317 explicit OptionalHIPStreamGuardMasqueradingAsCUDA() : guard_() {} in OptionalHIPStreamGuardMasqueradingAsCUDA() argument
318 explicit OptionalHIPStreamGuardMasqueradingAsCUDA(Stream stream) : guard_(stream) {} in OptionalHIPStreamGuardMasqueradingAsCUDA() argument
319 …explicit OptionalHIPStreamGuardMasqueradingAsCUDA(std::optional<Stream> stream_opt) : guard_(strea… in OptionalHIPStreamGuardMasqueradingAsCUDA() argument
322OptionalHIPStreamGuardMasqueradingAsCUDA& operator=(const OptionalHIPStreamGuardMasqueradingAsCUDA… argument
324OptionalHIPStreamGuardMasqueradingAsCUDA& operator=(OptionalHIPStreamGuardMasqueradingAsCUDA&& oth… argument
326 void reset_stream(Stream stream) { guard_.reset_stream(stream); } in reset_stream()
328 std::optional<HIPStreamMasqueradingAsCUDA> original_stream() const { in original_stream()
337 std::optional<HIPStreamMasqueradingAsCUDA> current_stream() const { in current_stream()
346 void reset() { guard_.reset(); } in reset()