/*
 * Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
 *
 * SPDX-License-Identifier: MIT
 */

#ifndef _OCL_PinnedBufferWriteSpeed_H_
#define _OCL_PinnedBufferWriteSpeed_H_

#include "OCLTestImp.h"

class OCLPerfPinnedBufferWriteSpeed : public OCLTestImp {
 public:
  OCLPerfPinnedBufferWriteSpeed();
  virtual ~OCLPerfPinnedBufferWriteSpeed();

 public:
  virtual void open(unsigned int test, char* units, double& conversion, unsigned int deviceID);
  virtual void run(void);
  virtual unsigned int close(void);

  static const unsigned int NUM_ITER = 1000;

  cl_context context_;
  cl_command_queue cmd_queue_;
  cl_mem inBuffer_;
  cl_mem outBuffer_;
  cl_int error_;

  unsigned int bufSize_;
  bool persistent;
  bool allocHostPtr;
  bool useHostPtr;
  unsigned int numIter;
  char* hostMem;
  char* alignedMem;
  size_t alignment;
  unsigned int offset;
  bool isAMD;
  char platformVersion[32];
};

class OCLPerfPinnedBufferWriteRectSpeed : public OCLPerfPinnedBufferWriteSpeed {
 public:
  OCLPerfPinnedBufferWriteRectSpeed() : OCLPerfPinnedBufferWriteSpeed() {}

 public:
  virtual void run(void);
};

#endif  // _OCL_PinnedBufferWriteSpeed_H_