exo icon indicating copy to clipboard operation
exo copied to clipboard

fix bfloat16 stuff

Open TanishkBansode opened this issue 11 months ago • 1 comments

  1. As numpy doesn't support bfloat16, updated code to check if it is bf16, and if yes, then convert to float32.

  2. also renamed the output_data and first_layer variables to have a better understanding what contains what( like output_data_mx and output_data_np).

TanishkBansode avatar May 12 '25 11:05 TanishkBansode

  1. As numpy doesn't support bfloat16, updated code to check if it is bf16, and if yes, then convert to float32.
  2. also renamed the output_data and first_layer variables to have a better understanding what contains what( like output_data_mx and output_data_np).

Related to #799 and #811

TanishkBansode avatar May 12 '25 11:05 TanishkBansode