Skip to content

[2.10] Support BF16 and FP16 - [MOD-6765, MOD-6776]#4674

Merged
GuyAv46 merged 1 commit into
2.10from
backport-4584-to-2.10
May 26, 2024
Merged

[2.10] Support BF16 and FP16 - [MOD-6765, MOD-6776]#4674
GuyAv46 merged 1 commit into
2.10from
backport-4584-to-2.10

Conversation

@GuyAv46

@GuyAv46 GuyAv46 commented May 26, 2024

Copy link
Copy Markdown
Collaborator

Backport of #4584 to 2.10.

  • adding support for the bf16 type in the API

  • support JSON casting to BF16

  • improved casting

  • add support for fp16

  • update VecSim tag (supports fp16 and bf16)

  • remove temporary definitions

  • added tests

  • minor fix

  • review fixes (tests)

  • more data types in multithread

  • fix another test

  • remove NaN/Inf support

  • replace bfloat16 module with tensorflow

  • limit urllib3 version

  • remove cpu only tensorflow

  • set jsons from numpy array

  • fix build for arm ubuntu

  • attempt

  • attempt

  • test

  • attempt

  • attempt

  • another attempt

  • implement bfloat16 myself

  • revert installation scripts

  • simplify create_np_array_typed

  • extend comment

* adding support for the bf16 type in the API

* support JSON casting to BF16

* improved casting

* add support for fp16

* update VecSim tag (supports fp16 and bf16)

* remove temporary definitions

* added tests

* minor fix

* review fixes (tests)

* more data types in multithread

* fix another test

* remove NaN/Inf support

* replace bfloat16 module with tensorflow

* limit urllib3 version

* remove cpu only tensorflow

* set jsons from numpy array

* fix build for arm ubuntu

* attempt

* attempt

* test

* attempt

* attempt

* another attempt

* implement bfloat16 myself

* revert installation scripts

* simplify create_np_array_typed

* extend comment
@GuyAv46 GuyAv46 requested a review from meiravgri May 26, 2024 08:20
@GuyAv46 GuyAv46 enabled auto-merge May 26, 2024 08:20
@codecov

codecov Bot commented May 26, 2024

Copy link
Copy Markdown

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 86.16%. Comparing base (e417057) to head (899d56e).

Additional details and impacted files
@@            Coverage Diff             @@
##             2.10    #4674      +/-   ##
==========================================
+ Coverage   86.14%   86.16%   +0.01%     
==========================================
  Files         190      190              
  Lines       34490    34531      +41     
==========================================
+ Hits        29713    29753      +40     
- Misses       4777     4778       +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@GuyAv46 GuyAv46 added this pull request to the merge queue May 26, 2024
@GuyAv46 GuyAv46 removed this pull request from the merge queue due to a manual request May 26, 2024
@GuyAv46 GuyAv46 added this pull request to the merge queue May 26, 2024
Merged via the queue into 2.10 with commit e803dd3 May 26, 2024
@GuyAv46 GuyAv46 deleted the backport-4584-to-2.10 branch May 26, 2024 10:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants