diff --git a/src/plugins/select/bluegene/runjob_plugin.cc b/src/plugins/select/bluegene/runjob_plugin.cc index afc3072dc3bb5484fc6daf871c06a495af910715..e85129856994ad9161bdf4a2e3ad3d02bb76d60d 100644 --- a/src/plugins/select/bluegene/runjob_plugin.cc +++ b/src/plugins/select/bluegene/runjob_plugin.cc @@ -94,6 +94,15 @@ Plugin::~Plugin() void Plugin::execute(bgsched::runjob::Verify& verify) { boost::lock_guard<boost::mutex> lock( _mutex ); + char *bg_block_id = getenv("MPIRUN_PARTITION"); + + if (!bg_block_id || (strlen(bg_block_id) < 3)) { + printf("YOU ARE OUTSIDE OF SLURM!!!!\n"); + verify.denyJob(DenyJob::Yes); + return; + } + + verify.block(bg_block_id); std::cout << "starting job from pid " << verify.pid() << std::endl; std::cout << "executable: " << verify.exe() << std::endl;