X-Git-Url: https://git.donarmstrong.com/?p=uiuc_igb_scripts.git;a=blobdiff_plain;f=dqsub;h=c0696c17a4ea4164cd34243b676240343e0b767e;hp=51546237f17bee6d6c0fbdbae53cdf99e4d27582;hb=HEAD;hpb=b0ca2cd2abbc503a304210db7f3f5fc277679532

diff --git a/dqsub b/dqsub
index 5154623..c0696c1 100755
--- a/dqsub
+++ b/dqsub
@@ -158,7 +158,7 @@ if ($options{interactive} and @ARGV) {
 }
 
 if (not defined $options{batch}) {
-    qx/which sbatch/;
+    qx{which sbatch >/dev/null 2>&1};
     if ($? == 0) {
         $options{batch} = 'slurm'
     } else {
@@ -194,7 +194,12 @@ if ($options{batch} eq 'pbs') {
 
 if ($options{interactive}) {
     print STDERR 'running: '.$JOB_SUBMITTER.' '.join(' ',@qsub_options) if $DEBUG;
-    exec($JOB_SUBMITTER,@qsub_options);
+    if ($options{batch} eq 'pbs') {
+        exec($JOB_SUBMITTER,@qsub_options);
+    } else {
+        exec('srun',@qsub_options,$ENV{SHELL}//'bash');
+    }
+
 } else {
     my @array = ();
     if ($options{array}) {
@@ -203,7 +208,12 @@ if ($options{interactive}) {
         if ($options{array_all_in_one_job}) {
             $options{array_per_job} = scalar @array;
         } else {
-            push @qsub_options,'-t','1-'. ceil(scalar @array / $options{array_per_job});
+            if ($options{batch} eq 'pbs') {
+                push @qsub_options,'-t';
+            } else {
+                push @qsub_options,'-a';
+            }
+            push @qsub_options,'1-'. ceil(scalar @array / $options{array_per_job});
             if ($options{array_slot_limit}) {
                 $qsub_options[$#qsub_options] .= '%'.$options{array_slot_limit};
             }
@@ -291,7 +301,7 @@ sub generate_slurm_options{
         push @qo,'--mem='.$options{mem};
     }
     if ($options->{interactive}) {
-        push @qo,'-I';
+        push @qo,'--pty';
     }
     if ($options->{name}) {
         push @qo,'-J',$options->{name};