Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def manipulator(self):
"""
Define the search space by creating a
ConfigurationManipulator
"""
# set the global flags needed for printing schedules
if self.args.enable_NUMA_tuning == 0:
self.enable_NUMA_tuning = False
if self.args.enable_parallel_tuning == 0:
self.enable_parallel_tuning = False
if self.args.enable_denseVertexSet_tuning == 0:
self.enable_denseVertexSet_tuning = False
manipulator = ConfigurationManipulator()
manipulator.add_parameter(
EnumParameter('direction',
['SparsePush','DensePull', 'SparsePush-DensePull', 'DensePush-SparsePush']))
if self.enable_parallel_tuning:
manipulator.add_parameter(EnumParameter('parallelization',['dynamic-vertex-parallel','edge-aware-dynamic-vertex-parallel']))
else:
manipulator.add_parameter(EnumParameter('parallelization', ['serial']))
manipulator.add_parameter(IntegerParameter('numSSG', 1, self.args.max_num_segments))
if self.enable_NUMA_tuning:
manipulator.add_parameter(EnumParameter('NUMA',['serial','static-parallel']))
if self.enable_denseVertexSet_tuning:
manipulator.add_parameter(EnumParameter('DenseVertexSet', ['boolean-array', 'bitvector']))
def manipulator(self):
"""
Define the search space by creating a
ConfigurationManipulator
"""
if not self.args.enable_NUMA_tuning == 1:
self.enable_NUMA_tuning = False
if not self.args.enable_parallel_tuning == 1:
self.enable_parallel_tuning = False
manipulator = ConfigurationManipulator()
manipulator.add_parameter(
EnumParameter('direction',
['SparsePush','DensePull', 'SparsePush-DensePull', 'DensePush-SparsePush']))
if self.enable_parallel_tuning:
manipulator.add_parameter(EnumParameter('parallelization',['dynamic-vertex-parallel','edge-aware-dynamic-vertex-parallel']))
else:
manipulator.add_parameter(EnumParameter('parallelization', ['serial']))
manipulator.add_parameter(IntegerParameter('numSSG', 1, 20))
if self.enable_NUMA_tuning:
manipulator.add_parameter(EnumParameter('NUMA',['serial','static-parallel']))
return manipulator
def manipulator(self):
"""
Define the search space by creating a
ConfigurationManipulator
"""
self.dimset = libxsmm_utilities.load_mnklist(self.args.mnk, 0, -1)
self.granularity = 1
assert(0 < self.granularity)
m_max = (64 + self.granularity - 1) / self.granularity
n_max = (256 + self.granularity - 1) / self.granularity
k_max = (256 + self.granularity - 1) / self.granularity
m_param = IntegerParameter("M", self.granularity, m_max)
n_param = IntegerParameter("N", self.granularity, n_max)
k_param = IntegerParameter("K", self.granularity, k_max)
manipulator = ConfigurationManipulator()
manipulator.add_parameter(m_param)
manipulator.add_parameter(n_param)
manipulator.add_parameter(k_param)
return manipulator
def manipulator(self):
"""
Define the search space by creating a
ConfigurationManipulator
"""
manipulator = ConfigurationManipulator()
manipulator.add_parameter(
IntegerParameter('BLOCK_SIZE', 1, 10))
return manipulator
def manipulator(self):
manipulator = ConfigurationManipulator()
for flag in NVCC_FLAGS + PTXAS_FLAGS + NVLINK_FLAGS:
manipulator.add_parameter(
EnumParameter(flag,
["on", "off"]))
for d in [NVCC_PARAMS, PTXAS_PARAMS]:
for flag, values in d.iteritems():
manipulator.add_parameter(EnumParameter(flag, values))
for param, pmin, pmax in NVCC_NUM_PARAMS + PTXAS_NUM_PARAMS:
manipulator.add_parameter(IntegerParameter(param, pmin, pmax))
return manipulator
def manipulator(self):
manipulator = ConfigurationManipulator()
for flag in NVCC_FLAGS + PTXAS_FLAGS + NVLINK_FLAGS:
manipulator.add_parameter(
EnumParameter(flag,
["on", "off"]))
for d in [NVCC_PARAMS, PTXAS_PARAMS]:
for flag, values in d.iteritems():
manipulator.add_parameter(EnumParameter(flag, values))
for param, pmin, pmax in NVCC_NUM_PARAMS + PTXAS_NUM_PARAMS:
manipulator.add_parameter(IntegerParameter(param, pmin, pmax))
return manipulator
def manipulator(self):
"""
Define the search space by creating a
ConfigurationManipulator
"""
manipulator = ConfigurationManipulator()
manipulator.add_parameter(
EnumParameter('direction',
['push','pull','hybrid_dense']))
return manipulator