AutoDock-GPU: AutoDock for GPUs and other accelerators
About
- OpenCL and Cuda accelerated version of AutoDock4.2.6. It leverages its embarrasingly parallelizable LGA by processing ligand-receptor poses in parallel over multiple compute units.
- The OpenCL version was developed in collaboration with TU-Darmstadt and is able to target CPU, GPU, and FPGA architectures.
- The Cuda version was developed in collaboration with Nvidia to run AutoDock-GPU on the Oak Ridge National Laboratory's (ORNL) Summit, and it included a batched ligand pipeline developed by Aaron Scheinberg from Jubilee Development.
Features
- Besides the legacy Solis-Wets local search method, AutoDock-GPU adds newly implemented local-search methods based on gradients of the scoring function. One of these methods, ADADELTA, has proven to increase significantly the docking quality in terms of RMSDs and scores.
- It targets platforms based on GPU as well as multicore CPU accelerators.
- Observed speedups of up to 4x (quad-core CPU) and 56x (GPU) over the original serial AutoDock 4.2 (Solis-Wets) on CPU. The Cuda version is currently even faster than the OpenCL version.
- A batched ligand pipeline to run virtual screenings on the same receptor (both OpenCL and Cuda)
Setup
| Operating system | CPU | GPU |
|---|---|---|
| CentOS 6.7 & 6.8 / Ubuntu 14.04 & 16.04 | Intel SDK for OpenCL 2017 | AMD APP SDK v3.0 / CUDA v8.0, v9.0, and v10.0 |
| macOS Catalina 10.15.1 | Apple / Intel | Apple / Intel Iris, Radeon Vega 64, Radeon VII |
Other environments or configurations likely work as well, but are untested.
Compilation
make DEVICE=<TYPE> NUMWI=<NWI>| Parameters | Description | Values |
|---|---|---|
<TYPE> |
Accelerator chosen | CPU, GPU, CUDA, OCLGPU |
<NWI> |
work-group/thread block size | 1, 2, 4, 8, 16, 32, 64, 128, 256 |
When DEVICE=GPU is chosen, the Makefile will automatically tests if it can compile Cuda succesfully. To override, use DEVICE=CUDA or DEVICE=OCLGPU. The cpu target is only supported using OpenCL.
Hints: The best work-group size depends on the GPU and workload. Try NUMWI=128 or NUMWI=64 for modern cards with the example workloads. On macOS, use NUMWI=1 for CPUs.
After successful compilation, the host binary autodock_<type>_<N>wi is placed under bin.
| Binary-name portion | Description | Values |
|---|---|---|
| <type> | Accelerator chosen | cpu, gpu |
| <N> | work-group/thread block size | 1, 2, 4, 8,16, 32, 64, 128, 256 |
Usage
Basic command
./bin/autodock_<type>_<N>wi \
-ffile <protein>.maps.fld \
-lfile <ligand>.pdbqt \
-nrun <nruns>| Mandatory options | Description | Value |
|---|---|---|
| -ffile | Protein file | <protein>.maps.fld |
| -lfile | Ligand file | <ligand>.pdbqt |
Example
./bin/autodock_gpu_64wi \
-ffile ./input/1stp/derived/1stp_protein.maps.fld \
-lfile ./input/1stp/derived/1stp_ligand.pdbqt \
-nrun 10By default the output log file is written in the current working folder. Examples of output logs can be found under examples/output.
Supported arguments
| Argument | Description | Default value |
|---|---|---|
| -nrun | # LGA runs | 1 |
| -nev | # Score evaluations (max.) per LGA run | 2500000 |
| -ngen | # Generations (max.) per LGA run | 27000 |
| -lsmet | Local-search method | sw (Solis-Wets) |
| -lsit | # Local-search iterations (max.) | 300 |
| -psize | Population size | 150 |
| -mrat | Mutation rate | 2 (%) |
| -crat | Crossover rate | 80 (%) |
| -lsrat | Local-search rate | 80 (%) |
| -trat | Tournament (selection) rate | 60 (%) |
| -resnam | Name for docking output log | "docking" |
| -hsym | Handle symmetry in RMSD calc. | 1 |
| -devnum | OpenCL/Cuda device number (counting starts at 1) | 1 |
| -cgmaps | Use individual maps for CG-G0 instead of the same one | 0 (use same map) |
| -heuristics | Ligand-based automatic search method and # evals | 0 |
| -heurmax | Asymptotic heuristics # evals limit (smooth limit) | 50000000 |
| -autostop | Automatic stopping criterion based on convergence | 0 |
| -asfreq | Autostop testing frequency (in # of generations) | 5 |
| -initswgens | Initial # generations of Solis-Wets instead of -lsmet | 0 |
| -filelist | Batch file | no default |
When the heuristics is used and -nev <max evals> is provided as a command line argument it provides the (hard) upper # of evals limit to the value the heuristics suggests. Conversely, -heurmax is the rolling-off type asymptotic limit to the heuristic's # of evals formula and should only be changed with caution.
The batch file is a text file containing the parameters to -ffile, -lfile, and -resnam each on an individual line. It is possible to only use one line to specify the Protein grid map file which means it will be used for all ligands. Here is an example:
./receptor1.maps.fld
./ligand1.pdbqt
Ligand 1
./receptor2.maps.fld
./ligand2.pdbqt
Ligand 2
./receptor3.maps.fld
./ligand3.pdbqt
Ligand 3
For a complete list of available arguments and their default values, check getparameters.cpp.
Documentation
Visit the project Wiki.