lecture-python.myst icon indicating copy to clipboard operation
lecture-python.myst copied to clipboard

Fix Long Run Time (bayes_nonconj)

Open mmcky opened this issue 2 years ago • 4 comments

It is currently taking the cache almost 3 hours to run this lecture set (from scratch).

There are a number of issues reported by tensor flow such as:

/__w/lecture-python.myst/lecture-python.myst/lectures/wealth_dynamics.md: Executing notebook using local CWD [mystnb]
2023-03-22 01:10:05.558920: E external/org_tensorflow/tensorflow/compiler/xla/service/slow_operation_alarm.cc:65] 
********************************
[Compiling module jit_wealth_time_series_for_loop_jax] Very slow compile?  If you want to file a bug, run with envvar XLA_FLAGS=--=/tmp/foo and attach the results.
********************************

which suggests jax isn't running as it should again.

  • [ ] once quantecon gpu machine is up and running we should test the lectures and jax on the 3080 to benchmark and diagnose what is going on.

mmcky avatar Mar 22 '23 01:03 mmcky