DeepLink-org / DIOPI

BSD 3-Clause "New" or "Revised" License
68 stars 34 forks source link

Align diopiStd with camb, torch_cuda, ascend for >=torch2.0 #1320

Closed DoorKickers closed 3 months ago

DoorKickers commented 3 months ago

Motivation and Context

torch.std has updated version since torch2.0, the changes is from

torch.std(input, dim, unbiased, keepdim=False, out=None)

to

torch.std(input, dim=None, correction=1, keepdim=False, out=None)

this pr is to update DIOPI related to support this new feature

Description

Use cases (Optional)

BC-breaking (Optional)

Checklist

Before PR:

After PR: