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.
-
inline explicit GpuPassStrategy(const GpuPassStrategy &other)¶
Construct by copying another GpuPassStrategy object.
- Parameters
other – [in] The GpuPassStrategy object we want to copy.
-
virtual void EnableCUDNN() override¶
Enable the use of cuDNN kernel.
-
virtual void EnableMKLDNN() override¶
Not supported in GPU mode yet.
-
virtual void EnableMkldnnQuantizer() override¶
Not supported in GPU mode yet.
-
virtual ~GpuPassStrategy() = default¶
Default destructor.
-
GpuPassStrategy()¶