File size: 1,002 Bytes
373e2bb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | # Kernel Configuration
OpenDDE has a safe PyTorch path and optional GPU kernels.
## LayerNorm
Default:
```bash
export LAYERNORM_TYPE=torch
```
Optional CUDA LayerNorm:
```bash
export LAYERNORM_TYPE=fast_layernorm
```
Use `torch` for CPU, debugging, or environments where CUDA extension compilation
is unavailable.
## Triangle kernels
Both triangle attention and triangle multiplication support:
| Value | Meaning |
| --- | --- |
| `auto` | Use cuEquivariance when available, otherwise PyTorch. |
| `cuequivariance` | Force cuEquivariance GPU kernels. |
| `torch` | Force PyTorch fallback. |
CLI flags:
```bash
opendde pred \
--triatt_kernel auto \
--trimul_kernel auto
```
## Compatibility run
```bash
LAYERNORM_TYPE=torch opendde pred \
-i examples/input.json \
-o ./output \
-n opendde_v1 \
--use_msa false \
--use_template false \
--use_rna_msa false \
--triatt_kernel torch \
--trimul_kernel torch \
--sample 1 \
--step 200 \
--cycle 10 \
--dtype fp32
```
|