Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

sync to head #771

Merged
merged 759 commits into from
Jul 15, 2024
Merged

sync to head #771

merged 759 commits into from
Jul 15, 2024

Conversation

shauryagup
Copy link
Collaborator

sync to head

maxtext authors and others added 30 commits April 11, 2024 12:41
Merge pull request #588 from google:vertex_flag
993f931 
PiperOrigin-RevId: 623909103
@NinaCai
Move tpu end-to-end test scripts to tpu folder
d676993
Merge pull request #575 from prrathi:main
a80c435 
PiperOrigin-RevId: 623946526
@NinaCai
Merge branch 'main' into nina/mv_tpu_end_to_end_scripts
0cdcdd1
Merge pull request #589 from google:nina/mv_tpu_end_to_end_scripts
28a3279 
PiperOrigin-RevId: 623965280
@michelle-yooh
unify WORKDIR to /deps
16a05c0
@chajath
Merge branch 'main' into yiinho-prebuilt-te
240e25d
@A9isha
Share GCS path between Gemma-7b tests
5b8a3c3
Merge pull request #583 from google:anisha-run-multicluster-test
3bba7a6 
PiperOrigin-RevId: 624288653
@michelle-yooh
Add README for llama2-7B
efb6d9e
Merge pull request #582 from google:yooh-llama2-7b-config-readme
71b78a1 
PiperOrigin-RevId: 624312270
Merge pull request #587 from google:yooh/fix_wordir
e947d62 
PiperOrigin-RevId: 624312415
@chajath
Merge remote-tracking branch 'origin/main' into yiinho-prebuilt-te
5d12cce
@chajath
Merge branch 'yiinho-prebuilt-te' of https://github.com/google/maxtext …
dd8fef5 
…into yiinho-prebuilt-te
Merge pull request #547 from google:yiinho-prebuilt-te
ebd39aa 
PiperOrigin-RevId: 624316618
@ssusie
adding script to fix the style and adding modified/fixed files with l…
9343eec 
…ine length 125
Merge pull request #596 from google:ssusie-pylint-125
3e316bd 
PiperOrigin-RevId: 625444223
@tonyjohnchen
Move apt install from rto_setup.sh to setup.sh
cee962b
Merge pull request #598 from google:rto
de9715d 
PiperOrigin-RevId: 625724059
@RoshaniN
Update instructions for installing snap.
57c9905
Merge pull request #601 from google:RoshaniN-patch-1
f52e6f7 
PiperOrigin-RevId: 626074066
@patemotter
Removes batch size from prefill attention calculation.
f9498fa
@patemotter
Fixes for inf testing.
8a8e642
@patemotter
Revert "Fixes for inf testing."
c2c3bce 
This reverts commit b15b1d5.
@patemotter
Fixes
6c355d7
@khatwanimohit
Fix subset of hosts dataloading
20f2a0d
Merge pull request #586 from google:mohit/subset_v4
6ec7556 
PiperOrigin-RevId: 626387060
@morgandu
inference microbenchmark
b46783c 
  - allow run specified stages
  - allow run specific prefill length(s)
  - delete prefill result
  - printout prefill result

added funcs in max_utils
@patemotter
Merge branch 'main' into patemotter_attn_calc_fix
2806017
Merge pull request #597 from google:mor--inference
0e1c078 
PiperOrigin-RevId: 626410509
maxtext authors and others added 26 commits July 2, 2024 20:09
Merge pull request #740 from google:emergency-restore
410dcad 
PiperOrigin-RevId: 648923148
Merge pull request #746 from google:yooh/convergence-test
290dc66 
PiperOrigin-RevId: 648942857
@RissyRan
Update tile size
4e13ea3
Merge pull request #747 from google:tile_size
3000396 
PiperOrigin-RevId: 649126836
@lukebaumann
Handle cases where memstats are not available for the device.
225f197 
Memstats are not guaranteed to be available and can throw an error or return None. This change will handle both `jaxlib.xla_extension.XlaRuntimeError` if the device is not a PjRt addressable device or `KeyError` if the memstats returns None if they are not available.
Merge pull request #739 from google:memstats
89edace 
PiperOrigin-RevId: 649176682
@RissyRan
Fix validation error for other models
eb5cf82
Merge pull request #750 from google:fix_mega_validation
b83a7a4 
PiperOrigin-RevId: 649248081
@vipannalla
Fix decode.py to also use first_token from prefill_call
85c105e
Merge pull request #756 from google:decode_prefill_fix
1b4cd15 
PiperOrigin-RevId: 650409674
@RissyRan
Add moe perf number
7534e5e
Merge pull request #751 from google:moe_result
5bc4029 
PiperOrigin-RevId: 650424212
Merge pull request #730 from google:smarter-sharding-parsing
a3d3260 
PiperOrigin-RevId: 650652469
@gobbleturk
move num_experts pyconfig assertion to fix tests
2cebd1e
Merge pull request #760 from google:mattdavidow-fix-assert
d0c70e9 
PiperOrigin-RevId: 650824599
@RissyRan
Cast type for inputs before kernel call
a26715e
Merge pull request #759 from google:moe_weight_type
84d8d05 
PiperOrigin-RevId: 650851189
@golechwierowicz
Move sharding overrides to models/ directory.
b506266 
PiperOrigin-RevId: 650994392
@RissyRan
Enable quantization for MoE Matmul implementation
6a0e570
Merge pull request #757 from google:moe_quantization
704ab1c 
PiperOrigin-RevId: 651070063
@dipannita08
Integrate and test Goodput Monitor with MaxText
f27f70a
Merge pull request #749 from google:integrate-goodput-monitor
0af4ee2 
PiperOrigin-RevId: 651091628
@tonyjohnchen
Adding Tokens/s/device to the log.
cae41ec
Merge pull request #761 from google:token_per_s
5f15eba 
PiperOrigin-RevId: 651113335
@singh-mitali
Adding support for mixed precision quantization configs.
8fb12fc
Merge pull request #762 from google:msingh-mp
8143e2d 
PiperOrigin-RevId: 651422435
@shauryagup shauryagup self-assigned this Jul 15, 2024
@shauryagup shauryagup marked this pull request as ready for review July 15, 2024 22:41
@shauryagup shauryagup requested a review from rwitten as a code owner July 15, 2024 22:41
@shauryagup shauryagup merged commit aac50a8 into shauryag-dev Jul 15, 2024
472 of 473 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@rwitten rwitten Awaiting requested review from rwitten

Assignees

@shauryagup shauryagup

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.