fix(Server.PyFR): Save number of GPUs in launch config

This is needed by PyFR workflow

Author: cjh1

server/taskflows/hpccloud/taskflow/pyfr.py

@@ -295,9 +295,9 @@ def setup_input(task, *args, **kwargs):
 
     # If we are running in the cloud determine backend to use
     if kwargs['cluster']['type'] == 'ec2':
-        machine_spec = kwargs.get('machine')
         # If we have GPUs use cuda
-        if int(machine_spec['gpu']) == 1:
+        gpu = parse('cluster.config.launch.gpu').find(kwargs)
+        if gpu and int(gpu[0].value) > 1:
             backend = {
                 'type': 'cuda',
                 'device-id': 'round-robin'

server/taskflows/hpccloud/taskflow/utility/__init__.py

@@ -50,7 +50,8 @@ def create_ec2_cluster(task, cluster, profile, ami):
         'master_instance_ami': ami,
         'node_instance_count': nodeCount,
         'node_instance_type': machine_type,
-        'node_instance_ami': ami
+        'node_instance_ami': ami,
+        'gpu': cluster['machine']['gpu']
     }
     provision_spec = 'gridengine/site'
     provision_params = {