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

Update notebooks #91

Merged
merged 1 commit into from
May 22, 2023
Merged

Update notebooks #91

merged 1 commit into from
May 22, 2023

Conversation

dachengx
Copy link
Collaborator

@dachengx dachengx commented May 22, 2023

Just update notebooks.

The test is under the environmental variables setting:

# export XLA_PYTHON_CLIENT_PREALLOCATE=false
unset XLA_PYTHON_CLIENT_PREALLOCATE
# export XLA_PYTHON_CLIENT_ALLOCATOR=platform
unset XLA_PYTHON_CLIENT_ALLOCATOR
# export DO_NOT_USE_APPROX_IN_BINOM=1
unset DO_NOT_USE_APPROX_IN_BINOM

unset DO_NOT_USE_APPROX_IN_BINOM will use Normal as an approximation of Binomial which will greatly increase run speed.

For XLA_PYTHON_CLIENT_PREALLOCATE and XLA_PYTHON_CLIENT_ALLOCATOR, one can check reference: https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html

@github-actions
Copy link

Pull Request Test Coverage Report for Build 5041890193

  • 0 of 0 changed or added relevant lines in 0 files are covered.
  • No unchanged relevant lines lost coverage.
  • Overall coverage remained the same at 80.159%

Totals Coverage Status
Change from base Build 5041849530: 0.0%
Covered Lines: 1818
Relevant Lines: 2268

💛 - Coveralls

@dachengx dachengx merged commit 1fe7648 into master May 22, 2023
7 checks passed
@dachengx dachengx deleted the update_nb branch May 22, 2023 04:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant