integrations/pjrt/src/iree_pjrt/cuda/client.h

// Copyright 2022 The IREE Authors
//
// Licensed under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#ifndef IREE_PJRT_PLUGIN_PJRT_CUDA_CLIENT_H_
#define IREE_PJRT_PLUGIN_PJRT_CUDA_CLIENT_H_

#include "iree/hal/drivers/cuda/api.h"
#include "iree_pjrt/common/api_impl.h"

namespace iree::pjrt::cuda {

class CUDAClientInstance final : public ClientInstance {
 public:
  CUDAClientInstance(std::unique_ptr<Platform> platform);
  ~CUDAClientInstance();
  iree_status_t CreateDriver(iree_hal_driver_t** out_driver) override;
  bool SetDefaultCompilerFlags(CompilerJob* compiler_job) override;

 private:
};

}  // namespace iree::pjrt::cuda

#endif  // IREE_PJRT_PLUGIN_PJRT_CPU_CLIENT_H_