Class PluginConfig

tensorflow/stream_executor/plugin.h:60–83

A PluginConfig describes the set of plugins to be used by a StreamExecutor instance. Each plugin is defined by an arbitrary identifier, usually best set to the address static member in the implementation (to avoid conflicts). A PluginConfig may be passed to the StreamExecutor constructor - the plugins described therein will be used to provide BLAS, DNN, FFT, and RNG functionality. Platform-approp

// targets. See the cuda, opencl and host BUILD files for implemented plugin
// support (search for "plugin").
class PluginConfig {
 public:
  // Value specifying the platform's default option for that plugin.
  static const PluginId kDefault;

  // Initializes all members to the default options.
  PluginConfig();

  bool operator==(const PluginConfig& rhs) const;

  // Sets the appropriate library kind to that passed in.
  PluginConfig& SetBlas(PluginId blas);
  PluginConfig& SetDnn(PluginId dnn);
  PluginConfig& SetFft(PluginId fft);
  PluginConfig& SetRng(PluginId rng);

  PluginId blas() const { return blas_; }
  PluginId dnn() const { return dnn_; }
  PluginId fft() const { return fft_; }
  PluginId rng() const { return rng_; }

 private:
  PluginId blas_, dnn_, fft_, rng_;
};

}  // namespace stream_executor