doifft_client.h

Go to the documentation of this file.

 
#ifndef DOIFFT_CLIENT_H_
#define DOIFFT_CLIENT_H_
 
#include "common_typedef_sdk.h"
#include "config.h"
#include "doer.h"
#include "memory_manage.h"
#include "mkl_dfti.h"
#include "stats.h"
#include "symbols.h"
 
class DoIFFTClient : public Doer {
 public:
  DoIFFTClient(Config* in_config, int in_tid,
               Table<complex_float>& in_ifft_buffer, char* in_socket_buffer,
               Stats* in_stats_manager);
  ~DoIFFTClient() override;
 
  EventData Launch(size_t tag) override;
 
 private:
  Table<complex_float>& ifft_buffer_;
  char* socket_buffer_;
  DurationStat* duration_stat_;
  DFTI_DESCRIPTOR_HANDLE mkl_handle_;
  float* ifft_out_;  // Buffer for IFFT output
  complex_float* ifft_shift_tmp_;
  float ifft_scale_factor_;
};
 
#endif  // DOIFFT_CLIENT_H_