Skip to content

conv_cudnn_op not utilizing workspace #6320

@tonyyang-svail

Description

@tonyyang-svail

The default value of workspace_size_MB has been set to 4096.

AddAttr<int>("workspace_size_MB",
"workspace size for cudnn, in MB, "
"workspace is a section of GPU memory which will be "
"allocated/freed each time the operator runs, larger "
"workspace size can increase performance but also requires "
"better hardware. This size should be chosen carefully.")
.SetDefault(4096);

workspace_size_MB is of type int. It will overflow at

if (user_workspace_size > 0) {
workspace_size_limit = user_workspace_size * 1024 * 1024;
}

The similar problem appears at backward.

The test won't fail because workspace of size 0 is still a valid initialization. If we fix this we can get 50% performance improvements on ConvNet on TitanX. #6321

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions