Skip to content
This repository has been archived by the owner on Nov 4, 2024. It is now read-only.

feat: use sleefpirates #101

Merged
merged 9 commits into from
Jul 25, 2024
Merged

feat: use sleefpirates #101

merged 9 commits into from
Jul 25, 2024

Conversation

avik-pal
Copy link
Member

@avik-pal avik-pal commented Jul 23, 2024 •
edited
Loading

fixes #99

Checklist

  • use SLEEFPirates in the forward pass
  • make enzyme work
    • Bilinear starts to fail not sure why
  • can we use SLEEFPirates in the backward pass as well?
  • add tests comparing the generic activations with the SLEEFPirates version

Enzyme seems still sad from downstream testing.

@avik-pal avik-pal force-pushed the ap/sleefpirates branch 3 times, most recently from 481cee2 to 74cbfde Compare July 23, 2024 06:28
@avik-pal avik-pal force-pushed the ap/sleefpirates branch 2 times, most recently from 99265ff to f0cf6c6 Compare July 24, 2024 01:30
@avik-pal avik-pal force-pushed the ap/sleefpirates branch 2 times, most recently from fb300e1 to f8ace04 Compare July 24, 2024 02:02
This was referenced Jul 24, 2024
Merged
Open
@avik-pal avik-pal merged commit 9b3a4d7 into main Jul 25, 2024
19 of 27 checks passed
@avik-pal avik-pal deleted the ap/sleefpirates branch July 25, 2024 02:52
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

Activation Functions using SLEEFPirates
1 participant
@avik-pal