cuFFTAdvisor/inputParser.h at master · I2PC/cuFFTAdvisor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#ifndef CUFFTADVISOR_INPUTPARSER_H_
#define CUFFTADVISOR_INPUTPARSER_H_

#include <climits>
#include <cstdio>
#include <cstdlib>
#include "utils.h"

namespace cuFFTAdvisor {

class InputParser {
 public:
  InputParser(int argc, char **argv);
  ~InputParser();
  bool reportUnparsed(FILE *stream);

  int x, y, z, n;
  int device;
  int maxSignalInc;
  int maxMemMB;
  bool allowTransposition;
  bool squareOnly;
  bool crop;
  Tristate::Tristate isBatched;
  Tristate::Tristate isFloat;
  Tristate::Tristate isForward;
  Tristate::Tristate isInPlace;
  Tristate::Tristate isReal;

 private:
  InputParser(InputParser &other); // prohibit copy

  void parseDims();
  void parseDevice();
  int parseMaxSignalInc();
  int parseMaxMemMB();
  bool parseAllowTransposition();
  bool parseSquareOnly();
  bool parseCrop();
  Tristate::Tristate parseIsReal();
  Tristate::Tristate parseIsFloat();
  Tristate::Tristate parseIsForward();
  Tristate::Tristate parseIsBatched();
  Tristate::Tristate parseIsInPlace();

  int argc;
  char **argv;
};

}  // namespace cuFFTAdvisor

#endif  // CUFFTADVISOR_INPUTPARSER_H_