Skip to content

Perform contractions using TBLIS#40

Open
fevangelista wants to merge 13 commits into
masterfrom
tblis
Open

Perform contractions using TBLIS#40
fevangelista wants to merge 13 commits into
masterfrom
tblis

Conversation

@fevangelista

@fevangelista fevangelista commented Jun 25, 2020

Copy link
Copy Markdown
Collaborator

This PR will add the ability to perform contractions using the TBLIS library. TBLIS is enabled at compilation by passing the arguments:

-DENABLE_TBLIS=ON -DTBLIS=<tblis installation directory>
  • Add TBLIS to CMakeFiles.txt
  • Implement contract() using TBLIS
  • Perform straight DGEMM calls using BLAS
  • Tests run

@jturney

jturney commented Jun 25, 2020

Copy link
Copy Markdown
Owner

Nice! One of our grad students has been playing with TBLIS, too, with very impressive performance benefits.

@jturney jturney left a comment

Copy link
Copy Markdown
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks really good!

@loriab

loriab commented Jun 26, 2020

Copy link
Copy Markdown
Collaborator

One morsel of cmake integration you might like to do now is to made a "component" of TBLIS in the ambit cmake. Would be something like replacing shared with tblis and @BUILD_SHARED_LIBS@ with @ENABLE_TBLIS@ in https://github.com/dgasmith/gau2grid/blob/master/cmake/gau2gridConfig.cmake.in#L48-L54 . Then in your personal psi4's you can add find_package(ambit ... COMPONENTS tblis) so that you only detect ambits with tblis.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants