Skip to content

Ra#98

Merged
jamesthesnake merged 7 commits intomainfrom
ra
Jul 21, 2023
Merged

Ra#98
jamesthesnake merged 7 commits intomainfrom
ra

Conversation

@jamesthesnake
Copy link
Copy Markdown
Owner

@jamesthesnake jamesthesnake commented Jul 21, 2023

📌 Checklist before creating the PR

  • I have created an issue for this PR for traceability
  • The title follows the standard format: [doc/gemini/tensor/...]: A concise description
  • I have added relevant tags if possible for us to better distinguish different PRs

🚨 Issue number

Link this PR to your issue with words like fixed to automatically close the linked issue upon merge

e.g. fixed #1234, closed #1234, resolved #1234

📝 What does this PR do?

Summarize your work here.
if you have any plots/diagrams/screenshots/tables, please attach them here.

💥 Checklist before requesting a review

  • I have linked my PR to an issue (instruction)
  • My issue clearly describes the problem/feature/proposal, with diagrams/charts/table/code if possible
  • I have performed a self-review of my code
  • I have added thorough tests.
  • I have added docstrings for all the functions/methods I implemented

⭐️ Do you enjoy contributing to Colossal-AI?

  • 🌝 Yes, I do.
  • 🌚 No, I don't.

Tell us more if you don't enjoy contributing to Colossal-AI.

CjhHa1 and others added 7 commits July 17, 2023 17:30
@CjhHa1
revise shardformer readme (hpcaitech#4246)
9a4842c
@binmakeswell
[example] add llama pretraining (hpcaitech#4257)
7ff11b5
@tiandiao123
[Kernels] added triton-implemented of self attention for colossal-ai (h…
4b97754 
…pcaitech#4241)

* added softmax kernel

* added qkv_kernel

* added ops

* adding tests

* upload tets

* fix tests

* debugging

* debugging tests

* debugging

* added

* fixed errors

* added softmax kernel

* clean codes

* added tests

* update tests

* update tests

* added attention

* add

* fixed pytest checking

* add cuda check

* fix cuda version

* fix typo
@ver217
[lazy] support init on cuda (hpcaitech#4269)
fc5cef2 
* [lazy] support init on cuda

* [test] update lazy init test

* [test] fix transformer version
[checkpointio] Sharded Optimizer Checkpoint for Gemini Plugin (hpcait…
c6f6005 
…ech#4302)

* sharded optimizer checkpoint for gemini plugin

* modify test to reduce testing time

* update doc

* fix bug when keep_gatherd is true under GeminiPlugin
@ver217
[ci] support testmon core pkg change detection (hpcaitech#4305)
02192a6
@jamesthesnake
Merge pull request #97 from hpcaitech/main
c9f729e 
raa
@jamesthesnake jamesthesnake merged commit 6a5ca19 into main Jul 21, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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.

5 participants

@jamesthesnake @CjhHa1 @binmakeswell @tiandiao123 @ver217