Merge branch 'main' of https://github.com/aqlaboratory/openfold

13f8f163 · zhuwenwen · a509a4c5 · b5fa2ba3 · 13f8f163 · 13f8f163
Commit 13f8f163 authored Apr 26, 2023 by zhuwenwen
20 changed files
--- a/openfold/utils/exponential_moving_average.py
+++ b/openfold/utils/exponential_moving_average.py
--- a/openfold/utils/feats.py
+++ b/openfold/utils/feats.py
--- a/openfold/utils/import_weights.py
+++ b/openfold/utils/import_weights.py
--- a/openfold/utils/kernel/__init__.py
+++ b/openfold/utils/kernel/__init__.py
--- a/openfold/utils/kernel/attention_core.py
+++ b/openfold/utils/kernel/attention_core.py
--- a/openfold/utils/kernel/csrc/compat.h
+++ b/openfold/utils/kernel/csrc/compat.h
+// modified from https://github.com/NVIDIA/apex/blob/master/csrc/compat.h
+
+#ifndef TORCH_CHECK
+#define TORCH_CHECK AT_CHECK
+#endif
+
+#ifdef VERSION_GE_1_3
+#define DATA_PTR data_ptr
+#else
+#define DATA_PTR data
+#endif
--- a/openfold/utils/kernel/csrc/softmax_cuda.cpp
+++ b/openfold/utils/kernel/csrc/softmax_cuda.cpp
--- a/openfold/utils/kernel/csrc/softmax_cuda_kernel.cu
+++ b/openfold/utils/kernel/csrc/softmax_cuda_kernel.cu
--- a/openfold/utils/kernel/csrc/softmax_cuda_stub.cpp
+++ b/openfold/utils/kernel/csrc/softmax_cuda_stub.cpp
--- a/openfold/utils/logger.py
+++ b/openfold/utils/logger.py
--- a/openfold/utils/loss.py
+++ b/openfold/utils/loss.py
--- a/openfold/utils/lr_schedulers.py
+++ b/openfold/utils/lr_schedulers.py
--- a/openfold/utils/precision_utils.py
+++ b/openfold/utils/precision_utils.py
--- a/openfold/utils/rigid_utils.py
+++ b/openfold/utils/rigid_utils.py
--- a/openfold/utils/script_utils.py
+++ b/openfold/utils/script_utils.py
--- a/openfold/utils/seed.py
+++ b/openfold/utils/seed.py
--- a/openfold/utils/superimposition.py
+++ b/openfold/utils/superimposition.py
--- a/openfold/utils/suppress_output.py
+++ b/openfold/utils/suppress_output.py
--- a/openfold/utils/tensor_utils.py
+++ b/openfold/utils/tensor_utils.py
--- a/openfold/utils/trace_utils.py
+++ b/openfold/utils/trace_utils.py