lib/coderunner/system_modules/helios.rb in coderunner-0.15.7 vs lib/coderunner/system_modules/helios.rb in coderunner-0.15.8

@@ -7,7 +7,20 @@
 			#super
 		#end
 		def max_ppn
 			16
 		end
+    def mpi_prog
+	    nodes, ppn = @nprocs.split(/x/)
+      nprocstot = nodes.to_i * ppn.to_i
+      "srun -n #{nprocstot}"
+    end 
+    def run_command
+  # 		"qsub #{batch_script_file}"
+      if (ENV['CODE_RUNNER_LAUNCHER'].size > 0 rescue false)
+        return %[#@preamble #{mpi_prog} #{executable_location}/#{executable_name} #{parameter_string} > #{output_file} 2> #{error_file}]
+      else
+        "#@preamble #{mpi_prog}  #{executable_location}/#{executable_name} #{parameter_string}"
+      end
+    end
 	end
 end