cluster-computing - SLURM - ジョブが使用している特定の CPU を特定するにはどうすればよいですか?

Question

クラスター (19 ノード、40 コア) で現在実行されているジョブを監視するためのツールに取り組んでいます。slurm キュー内の各ジョブが使用している特定の cpus を特定する方法はありますか? 「pidstat」、「mpstat」、および「ps -eFj」を使用してデータを取得しています。これは、特定のコアで実行されているプロセスを示していますが、それらのプロセス ID を Slurm が使用するジョブ ID に関連付ける方法がありません。「scontrol show job」は多くの情報を提供しますが、特定の CPU 割り当ては提供しません。これを行う方法はありますか？

データを収集するコードは次のとおりです。

#!/usr/bin/env python

import subprocess
import threading
import time

def scan():
  data = [[None, None, None] for i in range(19)]
  def mpstat(node):
    if(node == 1):
      output = subprocess.check_output(['mpstat', '-P', 'ALL', '1', '1'])
    else:
      output = subprocess.check_output(['ssh', 'node' + str(node), 'mpstat', '-P', 'ALL', '1', '1'])
    data[node - 1][0] = output
  def pidstat(node):
    if(node == 1):
      output = subprocess.check_output(['pidstat', '1', '1'])
    else:
      output = subprocess.check_output(['ssh', 'node' + str(node), 'pidstat', '1', '1'])
    data[node - 1][1] = output
  def ps(node):
    if(node == 1):
      output = subprocess.check_output(['ps', '-eFj'])
    else:
      output = subprocess.check_output(['ssh', 'node' + str(node), 'ps', '-eFj'])
    data[node - 1][2] = output
  threads = [[None, None, None] for i in range(19)]
  for node in range(1, 19 + 1):
    threads[node - 1][0] = threading.Thread(target=mpstat, args=(node,))
    threads[node - 1][0].start()
    threads[node - 1][1] = threading.Thread(target=pidstat, args=(node,))
    threads[node - 1][1].start()
    threads[node - 1][2] = threading.Thread(target=ps, args=(node,))
    threads[node - 1][2].start()
  while True:
    alive = [[not t.isAlive() for t in n]  for n in threads]
    alive = [t for n in alive for t in n]
    if(all(alive)):
      break
    time.sleep(1.0)
  return(data)