python_kernel.h

#include "scanner/api/kernel.h"
#include "scanner/api/op.h"
#include "scanner/util/memory.h"
#include "scanner/metadata.pb.h"

namespace scanner {

class PythonKernel : public StenciledBatchedKernel {
 public:
  PythonKernel(const KernelConfig& config,
               const std::string& op_name,
               const std::string& kernel_code,
               const bool can_batch,
               const bool con_stencil);

  ~PythonKernel();

  void new_stream(const std::vector<u8>& args) override;

  void execute(const StenciledBatchedElements& input_columns,
               BatchedElements& output_columns) override;

  void reset() override;

  void fetch_resources(proto::Result* result) override;

  void setup_with_resources(proto::Result* result) override;

 private:
  KernelConfig config_;
  DeviceHandle device_;
  bool can_batch_;
  bool can_stencil_;
  std::string op_name_;
  std::string process_name_;
  std::string send_pipe_name_;
  std::string recv_pipe_name_;
  std::string kernel_name_;
};

}