sample_kernel.h

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

#include <boost/python.hpp>
#include <boost/python/numpy.hpp>

namespace scanner {

class SampleKernel : public Kernel {
 public:
  SampleKernel(const KernelConfig& config, const std::string& kernel_str,
               const std::string& pickled_config);

  ~SamplehonKernel();

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

 private:
  KernelConfig config_;
  DeviceHandle device_;
};

}