Thanks to visit codestin.com
Credit goes to github.com

Skip to content

[Contrib][Openfold Triton] Use json instead of pickle#1900

Merged
crcrpar merged 7 commits into
NVIDIA:masterfrom
Aidyn-A:fix_serialization_in_openfold_triton
May 15, 2025
Merged

[Contrib][Openfold Triton] Use json instead of pickle#1900
crcrpar merged 7 commits into
NVIDIA:masterfrom
Aidyn-A:fix_serialization_in_openfold_triton

Conversation

@Aidyn-A
Copy link
Copy Markdown
Collaborator

@Aidyn-A Aidyn-A commented May 5, 2025

In this PR I propose to use json serialization instead of pickle, as the latter one is known to be a RCE vulnerable.

cc @crcrpar @arnowaczynski

Comment thread apex/contrib/openfold_triton/__init__.py Outdated
Comment thread apex/contrib/test/openfold_triton/test_sync_triton_auto_tune_cache_across_gpus.py Outdated
Co-Authored-By: Masaki Kozuki <[email protected]>
@crcrpar crcrpar merged commit e74a67b into NVIDIA:master May 15, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants