DeepLink-org / DIOPI

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

Fix some cuda impl related && Add log1p #1336

Closed DoorKickers closed 3 months ago

DoorKickers commented 3 months ago

Motivation and Context

this pr is to: 1.fix diopi cuda impl related so that it could be compiled normally. 2.add log1p on impl cuda that supports float32 and float64. 3.add log1p on impl torch that supports almost all input type.

Description

Use cases (Optional)

BC-breaking (Optional)

Checklist

Before PR:

After PR: