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.