Skip to content

DNNl MaxPool can't get nan result when input's value is nan #37

Closed
@XiaobingSuper

Description

@XiaobingSuper

DNNL MaxPool should get nan value when input values are all nan.

test case:

import torch
import intel_pytorch_extension
from torch._six import inf, nan
intel_pytorch_extension.core.enable_auto_dnnl()

input = torch.full([1, 1] + 2 * [3], nan).to('dpcpp')
output = F.max_pool2d(x, 3, 3)

Metadata

Metadata

Assignees

No one assigned

    Labels

    BugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions