v6d-io / v6d

vineyard (v6d): an in-memory immutable data manager. (Project under CNCF, TAG-Storage)
https://v6d.io
Apache License 2.0
818 stars 117 forks source link

Support the bfloat16 for pytorch modules. #1872

Closed dashanji closed 2 months ago

dashanji commented 2 months ago

What do these changes do?

As titled

Related issue number

Fixes https://github.com/v6d-io/v6d/issues/1871

sighingnow commented 2 months ago

Raise a warning before the casting, as it is not lossless.

dashanji commented 2 months ago

Raise a warning before the casting, as it is not lossless.

Done.