cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

added mapping for bf16 to torch::kBFloat16

Open Bogumil-Sapinski-Mobica opened this issue 4 months ago • 0 comments

Mapping for cutlass bf16 type to pytorch kBFloat16 type. After analyzing pytorch code this type might be not fully supported. In pytorch/test/cpp/api/support.h line 81 there is a comment informing that not all tensor operations are implemented for this type.

Bogumil-Sapinski-Mobica avatar Sep 27 '24 08:09 Bogumil-Sapinski-Mobica