vart_plkernel.hpp

Go to the documentation of this file.

/*
 * Copyright 2024-2026 Advanced Micro Devices Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include <any>
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include "base/vart_plkernel_impl_base.hpp"
#include "vart_device.hpp"
#include "vart_plkernel_types.hpp"
 
namespace vart {
 
class PLKernelImplBase;
 
class PLKernel {
 public:
  PLKernel(PLKernelImplType type,
           const std::string& kernel_name,
           std::string& json_data,
           std::shared_ptr<Device> device);
 
  PLKernel(std::shared_ptr<PLKernelImplBase> ptr);
 
  const std::shared_ptr<PLKernelImplBase>& get_pimpl_handle() const;
 
  template <typename... Args>
  void process(Args&&... args) {
    try {
      std::vector<std::any> any_args = {std::forward<Args>(args)...};
      pimpl->process(any_args);
    } catch (...) {
      /* Rethrow the caught exception */
      throw;
    }
  }
 
  int wait(unsigned int timeout);
 
  template <typename PLKernelAnyInfo>
  void set_config(PLKernelAnyInfo& info) {
    pimpl->set_config(info);
  }
 
  template <typename PLKernelAnyInfo>
  void get_config(PLKernelAnyInfo& info) {
    std::any any_info = info;
    pimpl->get_config_impl(any_info);
    info = std::any_cast<PLKernelAnyInfo>(any_info);
  }
 
 private:
  std::shared_ptr<PLKernelImplBase> pimpl;  
};
 
}  // namespace vart