ExecutorDeviceType.h File Reference

#include <ostream>

Include dependency graph for ExecutorDeviceType.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Enumerations
enum	ExecutorDeviceType { ExecutorDeviceType::CPU, ExecutorDeviceType::GPU }

Functions
std::ostream &	operator<< (std::ostream &os, const ExecutorDeviceType device_type)
std::string	toString (const ExecutorDeviceType device_type)

Enumeration Type Documentation

enum ExecutorDeviceType

strong

Enumerator
CPU
GPU

Definition at line 23 of file ExecutorDeviceType.h.

{ CPU, GPU };

Function Documentation

std::ostream& operator<<	(	std::ostream &	os,
		const ExecutorDeviceType	device_type
	)

Definition at line 21 of file ExecutorDeviceType.cpp.

References CHECK_LT.

                                                                       {
  constexpr size_t array_size{2};
  constexpr char const* strings[array_size]{"CPU", "GPU"};
  auto index = static_cast<size_t>(device_type);
  CHECK_LT(index, array_size);
  return os << strings[index];
}

std::string toString

(

const ExecutorDeviceType

device_type

)

Definition at line 29 of file ExecutorDeviceType.cpp.

                                                   {
  std::stringstream ss;
  ss << device_type;
  return ss.str();
}