def __init__(self, program, mem_bytes, stack_size, n_threads=4):
# program = list of instructions
self.program = program
self.memory = VMMemory(mem_bytes)
self.threads = {}
self.done = False
self.debug = False
#self.thread_pool = Pool(n_threads)
self.threads_to_kill = []
self.threads_to_add = []
self.memory_to_write = {}
self.highest_thread_id = MASTER_THREAD_ID
def __init__(self, program, mem_bytes, stack_size, n_threads=4):
# program = list of instructions
self.program = program
self.memory = VMMemory(mem_bytes)
self.threads = {}
self.done = False
self.debug = False
#self.thread_pool = Pool(n_threads)
self.threads_to_kill = []
self.threads_to_add = []
self.memory_to_write = {}
self.highest_thread_id = MASTER_THREAD_ID
class VMRuntime(object):
def __init__(self, program, mem_bytes, stack_size, n_threads=4):
# program = list of instructions
self.program = program
self.memory = VMMemory(mem_bytes)
self.threads = {}
self.done = False
self.debug = False
#self.thread_pool = Pool(n_threads)
self.threads_to_kill = []
self.threads_to_add = []
self.memory_to_write = {}
self.highest_thread_id = MASTER_THREAD_ID
def get_inst(self, pc):
return self.program[pc]
def run(self):
self.timestep = 0
while not self.done:
self.step()
self.timestep += 1
def step(self):
for thread in self.threads:
self.threads[thread].step(self)
for thread in self.threads_to_kill:
if self.debug:
print '[Runtime:%d] Killing thread %d' % (self.timestep,
thread)
del self.threads[thread]
self.threads_to_kill = []
for thread in self.threads_to_add:
if self.debug:
print '[Runtime:%d] Spawning thread %d' % (self.timestep,
thread.id)
self.threads[thread.id] = thread
self.threads_to_add = []
for idx in self.memory_to_write:
type, val = self.memory_to_write[idx]
self.memory.store(type, idx, val)
self.memory_to_write = {}
if len(self.threads) == 0:
if self.debug:
print '[Runtime:%d] Done' % self.timestep
self.done = True
def spawn_master(self):
master = VMThread(0, MASTER_THREAD_ID)
self.threads[0] = master
def command_kill_thread(self, id):
self.threads_to_kill.append(id)
def command_storemem(self, type, idx, val):
if idx in self.memory_to_write:
raise ValueError('Data race at idx %d' % idx)
self.memory_to_write[idx] = (type, val)
def command_spawn_thread(self, new_stack, pc):
self.highest_thread_id += 1
new_thread = VMThread(pc, self.highest_thread_id)
new_thread.state.stack = new_stack
self.threads_to_add.append(new_thread)
class VMRuntime(object):
def __init__(self, program, mem_bytes, stack_size, n_threads=4):
# program = list of instructions
self.program = program
self.memory = VMMemory(mem_bytes)
self.threads = {}
self.done = False
self.debug = False
#self.thread_pool = Pool(n_threads)
self.threads_to_kill = []
self.threads_to_add = []
self.memory_to_write = {}
self.highest_thread_id = MASTER_THREAD_ID
def get_inst(self, pc):
return self.program[pc]
def run(self):
self.timestep = 0
while not self.done:
self.step()
self.timestep += 1
def step(self):
for thread in self.threads:
self.threads[thread].step(self)
for thread in self.threads_to_kill:
if self.debug:
print '[Runtime:%d] Killing thread %d' % (self.timestep, thread)
del self.threads[thread]
self.threads_to_kill = []
for thread in self.threads_to_add:
if self.debug:
print '[Runtime:%d] Spawning thread %d' % (self.timestep, thread.id)
self.threads[thread.id] = thread
self.threads_to_add = []
for idx in self.memory_to_write:
type, val = self.memory_to_write[idx]
self.memory.store(type, idx, val)
self.memory_to_write = {}
if len(self.threads) == 0:
if self.debug:
print '[Runtime:%d] Done' % self.timestep
self.done = True
def spawn_master(self):
master = VMThread(0, MASTER_THREAD_ID)
self.threads[0] = master
def command_kill_thread(self, id):
self.threads_to_kill.append(id)
def command_storemem(self, type, idx, val):
if idx in self.memory_to_write:
raise ValueError('Data race at idx %d' % idx)
self.memory_to_write[idx] = (type, val)
def command_spawn_thread(self, new_stack, pc):
self.highest_thread_id += 1
new_thread = VMThread(pc, self.highest_thread_id)
new_thread.state.stack = new_stack
self.threads_to_add.append(new_thread)
