v6d icon indicating copy to clipboard operation
v6d copied to clipboard

Implement the torch_tensor_builder and torch_tensor_resolver to support bfloat16

Open dashanji opened this issue 1 year ago • 0 comments

Describe your problem

Currently, the torch_tensor_builder and torch_tensor_resolver are cast to numpy builder and resolver directly, and bfloat16 is cast to float16 which caused the loss of accuracy.

dashanji avatar May 10 '24 03:05 dashanji