Function float_array_to_float8e5m2

src/test.cpp:62–68

}

std::vector<f8e5m2_t> float_array_to_float8e5m2(const std::vector<float>& data) {
  std::vector<f8e5m2_t> float8e5m2_data(data.size());
  for (size_t i = 0; i < data.size(); i++) {
    float8e5m2_data[i] = float_to_float8e5m2(data[i]);
  }
  return float8e5m2_data;
}

void test_attn() {
  constexpr int TEST_SEQ_LEN = 4;