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.

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

Criteo workload variants #568

Merged
merged 161 commits into from
Dec 7, 2023
Merged
Changes from 1 commit
Commits
Show all changes
161 commits
Select commit Hold shift + click to select a range
3933a34
criteo variants
priyakasimbeg Nov 10, 2023
5f40758
add workload variants
priyakasimbeg Nov 10, 2023
38821bb
add criteo variants to workload registry
priyakasimbeg Nov 11, 2023
f28ecbc
workload registry
priyakasimbeg Nov 11, 2023
3e44f69
add regression tests
priyakasimbeg Nov 11, 2023
a3a827c
add empty line
priyakasimbeg Nov 11, 2023
e1e9b4e
formatting
priyakasimbeg Nov 11, 2023
d3334f5
regression test fix
priyakasimbeg Nov 13, 2023
b6443f2
add criteo_variants to valid workloads to startup.sh"
priyakasimbeg Nov 13, 2023
4805423
syntax fix
priyakasimbeg Nov 14, 2023
b5c1979
add bsz for criteo1tb variants
priyakasimbeg Nov 14, 2023
87a0425
change regression test name for variants
priyakasimbeg Nov 14, 2023
6e6e786
syntax fix
priyakasimbeg Nov 14, 2023
dc208d0
lint fix
priyakasimbeg Nov 14, 2023
2ee0f77
formatting
priyakasimbeg Nov 14, 2023
fa75663
fix
priyakasimbeg Nov 14, 2023
d21f05b
fix typo
priyakasimbeg Nov 14, 2023
8776748
fix resnet
priyakasimbeg Nov 15, 2023
6c8fbe9
modify regresion test
priyakasimbeg Nov 15, 2023
7b7e750
modify_test
priyakasimbeg Nov 15, 2023
e601db4
add tests
priyakasimbeg Nov 16, 2023
7a71040
Merge branch 'dev' into criteo_workload_variants
priyakasimbeg Nov 16, 2023
8d2e828
regression tests
priyakasimbeg Nov 16, 2023
2193830
remove variant bsz
priyakasimbeg Nov 16, 2023
d46d5e0
add helper fn for get_baseworkload_name
priyakasimbeg Nov 16, 2023
7ddffd3
fix
priyakasimbeg Nov 16, 2023
6b2ed1f
modify conformer_resnet model
priyakasimbeg Nov 16, 2023
cd2d672
ln
priyakasimbeg Nov 16, 2023
f0a369a
fix
priyakasimbeg Nov 16, 2023
01d668c
fix
priyakasimbeg Nov 16, 2023
910e974
fix
priyakasimbeg Nov 16, 2023
063c229
debugging
priyakasimbeg Nov 16, 2023
2bf1988
fix
priyakasimbeg Nov 16, 2023
2c2a7a9
debugging
priyakasimbeg Nov 16, 2023
e698bc7
resnet block
priyakasimbeg Nov 17, 2023
9fd4efa
add resnet block to criteo resnet variant
priyakasimbeg Nov 17, 2023
bc78a19
add back dims
priyakasimbeg Nov 17, 2023
b1d2224
fix
priyakasimbeg Nov 17, 2023
6a46a6c
resnet
priyakasimbeg Nov 17, 2023
83c4ade
comment out pytorch
priyakasimbeg Nov 17, 2023
e60a159
resnet fix'
priyakasimbeg Nov 17, 2023
e95fe17
fix
priyakasimbeg Nov 17, 2023
1e5c709
fix
priyakasimbeg Nov 17, 2023
04da955
debugging
priyakasimbeg Nov 17, 2023
0b3c7a9
debugging
priyakasimbeg Nov 17, 2023
76169a2
mlp dims
priyakasimbeg Nov 17, 2023
cf7f221
variant fix
priyakasimbeg Nov 17, 2023
6fe0901
fix dlrm variant
priyakasimbeg Nov 17, 2023
0723d82
dlrm fix
priyakasimbeg Nov 17, 2023
d56269b
debugging
priyakasimbeg Nov 17, 2023
61f052b
debugging
priyakasimbeg Nov 17, 2023
ff5aa91
debug
priyakasimbeg Nov 17, 2023
df48937
debug
priyakasimbeg Nov 17, 2023
8a574e3
debug
priyakasimbeg Nov 17, 2023
82e42e3
debug
priyakasimbeg Nov 17, 2023
f2beed9
debug
priyakasimbeg Nov 17, 2023
17c1190
fix
priyakasimbeg Nov 17, 2023
e5a31f5
debug
priyakasimbeg Nov 17, 2023
7c98352
debuggingg
priyakasimbeg Nov 17, 2023
c9facb2
debugging
priyakasimbeg Nov 17, 2023
549be97
debugging
priyakasimbeg Nov 17, 2023
6967445
debug
priyakasimbeg Nov 17, 2023
5d237eb
debug
priyakasimbeg Nov 17, 2023
9a132b1
debugging
priyakasimbeg Nov 17, 2023
89dea72
debugging
priyakasimbeg Nov 18, 2023
2c54700
Merge branch 'criteo_workload_variants' of github.com:mlcommons/algor…
priyakasimbeg Nov 18, 2023
fdca6c5
debugging
priyakasimbeg Nov 18, 2023
29835e3
debugging
priyakasimbeg Nov 18, 2023
3a859eb
clarify output of diff test
priyakasimbeg Nov 18, 2023
7829863
key transform
priyakasimbeg Nov 18, 2023
a47e6b1
syntaxl
priyakasimbeg Nov 18, 2023
f75944d
diff
priyakasimbeg Nov 18, 2023
ef68ba8
logging
priyakasimbeg Nov 18, 2023
e096a3a
debug
priyakasimbeg Nov 18, 2023
0c7a864
debug
priyakasimbeg Nov 18, 2023
271412d
debug
priyakasimbeg Nov 18, 2023
650ef43
debugging
priyakasimbeg Nov 18, 2023
ed1cdba
debug
priyakasimbeg Nov 18, 2023
f59b48d
debug
priyakasimbeg Nov 18, 2023
47e7166
debug
priyakasimbeg Nov 18, 2023
39a6c31
debug
priyakasimbeg Nov 18, 2023
f70dd49
debug
priyakasimbeg Nov 18, 2023
0966015
debug
priyakasimbeg Nov 18, 2023
d7d4638
remove some debugging statements
priyakasimbeg Nov 18, 2023
b1c35d4
add debugging statement
priyakasimbeg Nov 18, 2023
e7e52f0
resnet fix
priyakasimbeg Nov 20, 2023
55d72d8
fix
priyakasimbeg Nov 20, 2023
ca8a00a
debugging
priyakasimbeg Nov 20, 2023
d929d76
compare_fix
priyakasimbeg Nov 20, 2023
5192df7
fix
priyakasimbeg Nov 20, 2023
a88f516
fix
priyakasimbeg Nov 20, 2023
7aca234
fix
priyakasimbeg Nov 20, 2023
c2d288e
block count
priyakasimbeg Nov 21, 2023
1bb484f
fix
priyakasimbeg Nov 21, 2023
b50e9dd
fix resnet jax
priyakasimbeg Nov 21, 2023
c83bdad
remove debugging statemetns
priyakasimbeg Nov 21, 2023
a6f2ba0
fix logging
priyakasimbeg Nov 21, 2023
b641ab9
add back print statemetns
priyakasimbeg Nov 21, 2023
62a43c8
resnet fix
priyakasimbeg Nov 21, 2023
84463e2
change block structures
priyakasimbeg Nov 21, 2023
6385365
fix
priyakasimbeg Nov 21, 2023
a2b64e9
debugging
priyakasimbeg Nov 22, 2023
ce7fcd9
debug
priyakasimbeg Nov 22, 2023
26f7864
fix
priyakasimbeg Nov 22, 2023
8243820
debug
priyakasimbeg Nov 22, 2023
c44b5bd
debug
priyakasimbeg Nov 22, 2023
6913e92
fix
priyakasimbeg Nov 22, 2023
6fc1830
debugigng
priyakasimbeg Nov 22, 2023
a140fc0
fix
priyakasimbeg Nov 22, 2023
9550c15
debug
priyakasimbeg Nov 22, 2023
bcf68ed
fix test
priyakasimbeg Nov 22, 2023
85be241
remove debugging statements
priyakasimbeg Nov 22, 2023
987686c
add jax model summary helper fn
priyakasimbeg Nov 22, 2023
c6e26da
formatting
priyakasimbeg Nov 22, 2023
90e8c80
update jax criteo workload
priyakasimbeg Nov 22, 2023
e440208
criteo workload variant
priyakasimbeg Nov 22, 2023
626420c
fixes
priyakasimbeg Nov 22, 2023
e070fe3
debug
priyakasimbeg Nov 22, 2023
412e4fd
debug
priyakasimbeg Nov 22, 2023
c982326
debug
priyakasimbeg Nov 22, 2023
f960a0a
debugging
priyakasimbeg Nov 22, 2023
122fbf4
debugging
priyakasimbeg Nov 22, 2023
bf12d5b
shape debugging
priyakasimbeg Nov 22, 2023
a234a8a
debugging
priyakasimbeg Nov 22, 2023
456573a
debugging
priyakasimbeg Nov 22, 2023
12f5da6
debugging
priyakasimbeg Nov 22, 2023
438deaf
debugging
priyakasimbeg Nov 22, 2023
49b45f6
debug
priyakasimbeg Nov 22, 2023
3ed3633
criteo variants
priyakasimbeg Nov 22, 2023
834ac9c
debugging
priyakasimbeg Nov 22, 2023
e4c7b34
fix
priyakasimbeg Nov 22, 2023
ac6baf7
embedding initialization criteo
priyakasimbeg Nov 22, 2023
e138c83
test embedding init
priyakasimbeg Nov 22, 2023
132223b
fix
priyakasimbeg Nov 22, 2023
071ddf3
add embedding init multiplier
priyakasimbeg Nov 22, 2023
134d0bb
debugging
priyakasimbeg Nov 22, 2023
955b70d
test
priyakasimbeg Nov 22, 2023
dadd4d6
fix
priyakasimbeg Nov 22, 2023
8e832a3
fix
priyakasimbeg Nov 22, 2023
07daeee
debug
priyakasimbeg Nov 22, 2023
263108f
debug
priyakasimbeg Nov 22, 2023
a7dd161
fix
priyakasimbeg Nov 22, 2023
40a4cde
debugging
priyakasimbeg Nov 22, 2023
1740325
debugging
priyakasimbeg Nov 22, 2023
b05f510
fix
priyakasimbeg Nov 22, 2023
3cf57a7
add tests
priyakasimbeg Nov 22, 2023
123dec9
Merge branch 'dev' into criteo_workload_variants
priyakasimbeg Nov 22, 2023
a5568ed
clean up;
priyakasimbeg Nov 22, 2023
76d1749
formatting
priyakasimbeg Nov 22, 2023
284c30c
reformat
priyakasimbeg Nov 22, 2023
547e225
sorting imports
priyakasimbeg Nov 22, 2023
a37d3fa
fix
priyakasimbeg Nov 22, 2023
1af2a8a
remove unused imports
priyakasimbeg Nov 22, 2023
72ede87
pylint
priyakasimbeg Nov 22, 2023
f6025a2
add exception
priyakasimbeg Nov 22, 2023
f947e23
add clarifying docs for submission example
priyakasimbeg Nov 22, 2023
765c363
formatting
priyakasimbeg Nov 22, 2023
9da4996
make exception specific
priyakasimbeg Nov 22, 2023
96bcea6
formatting
priyakasimbeg Nov 22, 2023
59384c6
pylint
priyakasimbeg Nov 22, 2023
c07e0e7
fixes
priyakasimbeg Dec 7, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
syntax fix
priyakasimbeg committed Nov 14, 2023
commit 480542387c3acb5af8d92f74ccc23524fd0bc34b
4 changes: 2 additions & 2 deletions algorithmic_efficiency/workloads/workloads.py
Original file line number Diff line number Diff line change
@@ -19,15 +19,15 @@
'criteo1tb_test': {
'workload_path': 'criteo1tb/criteo1tb',
'workload_class_name': 'Criteo1TbDlrmSmallTestWorkload',
}
},
'criteo1tb_layernorm': {
'workload_path': 'criteo1tb/criteo1tb',
'workload_class_name': 'Criteo1TbDlrmSmallLayerNormWorkload'
},
'criteo1tb_resnet': {
'workload_path': 'criteo1tb/criteo1tb',
'workload_class_name': 'Criteo1TbDlrmSmallResNetWorkload'
}
},
'fastmri': {
'workload_path': 'fastmri/fastmri',
'workload_class_name': 'FastMRIWorkload',