Add additional aliases for parallelization flags.

aiidateam · Sep 22, 2020 · 1a41823 · 1a41823
1 parent 56650fd
commit 1a41823
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/aiida_quantumespresso/calculations/__init__.py b/aiida_quantumespresso/calculations/__init__.py
@@ -57,11 +57,11 @@ class BasePwCpInputGenerator(CalcJob):
  )
  )
  _PARALLELIZATION_FLAG_ALIASES = {
- 'nimage': ['ni'],
- 'npools': ['nk'],
- 'nband': ['nb'],
- 'ntg': ['nt'],
- 'ndiag': ['northo', 'nd']
+ 'nimage': ['ni', 'nimages'],
+ 'npools': ['nk', 'npool'],
+ 'nband': ['nb', 'nbgrp', 'nband_group'],
+ 'ntg': ['nt', 'ntask_groups'],
+ 'ndiag': ['northo', 'nd', 'nproc_diag', 'nproc_ortho']
  }
 
  # Additional files that should always be retrieved for the specific plugin