Skip to content
This repository has been archived by the owner on Aug 26, 2022. It is now read-only.

implement tp-3d layers, wrapper, test codes and refactor all tp test codes and layers #133

Merged
merged 11 commits into from Jun 15, 2022

Conversation

bzantium
Copy link
Collaborator

  • implement tp-3d wrapper
  • rank transpose problem (tensor_3d_input_rank <-> tensor_3d_output_rank) by implementing ranking transpose function.
  • revise tp-3d layers for huggingface compatibility
  • implement tp-3d test codes
  • refactor all tp test codes
  • unify format across all tensor parallel modules.

@hyunwoongko
Copy link
Contributor

pre commit was failed 😭

@bzantium
Copy link
Collaborator Author

pre commit was failed 😭

Oops! I just applied precommit and pushed again! Thanks!

@hyunwoongko hyunwoongko merged commit 23f8fdf into master Jun 15, 2022
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
2 participants