Skip to content
Merged
Changes from 1 commit
Commits
Show all changes
48 commits
Select commit Hold shift + click to select a range
2e944b0
First batch of tf methods (to be continued)
ncassereau Nov 29, 2021
1653b70
Second batch of method (yet to debug)
ncassereau Nov 30, 2021
2fe3441
tensorflow for cpu
ncassereau Nov 30, 2021
5955643
add tf requirement
ncassereau Nov 30, 2021
d988df0
pep8 + bug
ncassereau Nov 30, 2021
86cdbc6
small changes
ncassereau Dec 1, 2021
93bc691
attempt to solve pymanopt bug with tf2
ncassereau Dec 1, 2021
8baeae9
attempt #2
ncassereau Dec 1, 2021
1881f85
attempt #3
ncassereau Dec 1, 2021
249ea2f
attempt 4
ncassereau Dec 1, 2021
b887746
docstring
ncassereau Dec 1, 2021
2d90386
Merge branch 'master' into tensorflow_backend
ncassereau Dec 3, 2021
a34aeff
correct pep8 violation introduced in merge conflicts resolution
ncassereau Dec 3, 2021
eae1c9a
attempt 5
ncassereau Dec 3, 2021
01fce56
attempt 6
ncassereau Dec 3, 2021
8223e76
just a random try
ncassereau Dec 3, 2021
aaac0ee
Revert "just a random try"
ncassereau Dec 3, 2021
0feea71
GPU tests for tensorflow
ncassereau Dec 6, 2021
32c2838
pep8
ncassereau Dec 6, 2021
3cadd11
attempt to solve issue with m2r2
ncassereau Dec 6, 2021
aaa7e4a
Remove transpose backend method
ncassereau Dec 6, 2021
2d77a37
Merge branch 'master' into tensorflow_backend
ncassereau Dec 6, 2021
245a3c2
first draft of benchmarker (need to correct time measurement)
ncassereau Dec 6, 2021
f269fda
prettier bench table
ncassereau Dec 7, 2021
bdba755
Bitsize and prettier device methods
ncassereau Dec 7, 2021
30e7ba7
prettified table bench
ncassereau Dec 7, 2021
689ae01
Bug corrected (results were mixed up in the final table)
ncassereau Dec 7, 2021
1347387
Better perf counter (for GPU support)
ncassereau Dec 7, 2021
96488ce
pep8
ncassereau Dec 7, 2021
59ea42e
EMD bench
ncassereau Dec 7, 2021
22c4d0c
solve bug if no GPU available
ncassereau Dec 7, 2021
eca3f80
pep8
ncassereau Dec 7, 2021
aa5257a
warning about tensorflow numpy api being required in the backend.py d…
ncassereau Dec 7, 2021
1f62660
Bug solve in backend docstring
ncassereau Dec 7, 2021
0f1d299
not covering code which requires a GPU
ncassereau Dec 7, 2021
0984000
Tensorflow gradients manipulation tested
ncassereau Dec 7, 2021
9e56ccd
Number of warmup runs is now customizable
ncassereau Dec 7, 2021
84fb002
typo
ncassereau Dec 7, 2021
9a7fd03
Remove some warnings while building docs
ncassereau Dec 7, 2021
acc9474
Change prettier_device to device_type in backend
ncassereau Dec 7, 2021
0593f87
Correct JAX mistakes preventing to see the CPU if a GPU is present
ncassereau Dec 7, 2021
321597f
Attempt to solve JAX bug in case no GPU is found
ncassereau Dec 7, 2021
ec72f30
Reworked benchmarks order and results storage & clear GPU after usage…
ncassereau Dec 8, 2021
9a80c7a
Add bench to backend docstring
ncassereau Dec 8, 2021
21d34ea
better benchs
ncassereau Dec 8, 2021
c507d3b
remove useless stuff
ncassereau Dec 8, 2021
d02f71a
Better device_type
ncassereau Dec 8, 2021
86faaa4
Now using MYST_PARSER and solving links issue in the README.md / onli…
ncassereau Dec 9, 2021
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
pep8
  • Loading branch information
ncassereau committed Dec 7, 2021
commit eca3f804b79cfdb842d865abdf7cbd1c66579034
2 changes: 1 addition & 1 deletion ot/backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -1644,7 +1644,7 @@ def _bench(self, callable, *args, n_runs=1):
start = time.perf_counter()
for _ in range(n_runs):
callable(*inputs)
if self.prettier_device(type_as) == "GPU":
if self.prettier_device(type_as) == "GPU":
end.record()
torch.cuda.synchronize()
duration = start.elapsed_time(end) / 1000.
Expand Down