Class GpuPassStrategy¶
Defined in File paddle_pass_builder.h
Inheritance Relationships¶
Base Type¶
public paddle::PassStrategy
(Class PassStrategy)
Class Documentation¶
-
class
paddle
::
GpuPassStrategy
: public paddle::PassStrategy¶ The GPU passes controller, it is used in AnalysisPredictor with GPU mode.
Public Functions
-
GpuPassStrategy
()¶ Default constructor of GpuPassStrategy.
-
GpuPassStrategy
(const GpuPassStrategy &other)¶ Construct by copying another GpuPassStrategy object.
- Parameters
[in] other
: The GpuPassStrategy object we want to copy.
-
void
EnableCUDNN
() override¶ Enable the use of cuDNN kernel.
-
void
EnableMKLDNN
() override¶ Not supported in GPU mode yet.
-
void
EnableMkldnnQuantizer
() override¶ Not supported in GPU mode yet.
-
~GpuPassStrategy
() = default¶ Default destructor.
-