Class GpuPassStrategy

Inheritance Relationships

Base Type

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

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.