HIPStreamMasqueradingAsCUDA Class — pytorch Architecture

Source Code

aten/src/ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h lines 11–25

class HIPStreamMasqueradingAsCUDA final : public c10::cuda::CUDAStream {
public:
  using c10::cuda::CUDAStream::CUDAStream;

  static HIPStreamMasqueradingAsCUDA unpack3(StreamId stream_id,
                                             DeviceIndex device_index,
                                             c10::DeviceType device_type) {
    // NB: constructor manages CUDA->HIP translation for us
    return HIPStreamMasqueradingAsCUDA(Stream::unpack3(
        stream_id, device_index, device_type));
  }

  // New method, gets the underlying "HIPStream" [CUDAStream]
  c10::cuda::CUDAStream hip_stream() const { return *this; }
};