/* Translate a job's feature specification to node boot options
* RET node boot options, must be xfreed */
extern char *node_features_g_job_xlate(char *job_features)
{
DEF_TIMERS;
char *node_features = NULL, *tmp_str;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
tmp_str = (*(ops[i].job_xlate))(job_features);
if (tmp_str) {
if (node_features) {
xstrfmtcat(node_features, ",%s", tmp_str);
xfree(tmp_str);
} else {
node_features = tmp_str;
}
}
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_job_xlate");
return node_features;
}
/* Translate a node's feature specification by replacing any features associated
* with this plugin in the original value with the new values, preserving any
* features that are not associated with this plugin
* RET node's new merged features, must be xfreed */
extern char *node_features_g_node_xlate(char *new_features, char *orig_features)
{
DEF_TIMERS;
char *new_value = NULL, *tmp_str;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
if (new_value)
tmp_str = new_value;
else if (orig_features)
tmp_str = xstrdup(orig_features);
else
tmp_str = NULL;
new_value = (*(ops[i].node_xlate))(new_features, tmp_str);
xfree(tmp_str);
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_node_xlate");
return new_value;
}
/* Get node features plugin configuration */
extern List node_features_g_get_config(void)
{
DEF_TIMERS;
int i, rc;
List conf_list = NULL;
config_plugin_params_t *p;
START_TIMER;
rc = node_features_g_init();
if (g_context_cnt > 0)
conf_list = list_create(destroy_config_plugin_params);
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && (rc == SLURM_SUCCESS)); i++) {
p = xmalloc(sizeof(config_plugin_params_t));
p->key_pairs = list_create(destroy_config_key_pair);
(*(ops[i].get_config))(p);
if (!p->name)
destroy_config_plugin_params(p);
else
list_append(conf_list, p);
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_get_config");
return conf_list;
}
/* Return count of node_feature plugins configured */
extern int node_features_g_count(void)
{
int rc;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
rc = g_context_cnt;
slurm_mutex_unlock(&g_context_lock);
return rc;
}
/* Get this node's current and available MCDRAM and NUMA settings from BIOS.
* avail_modes IN/OUT - available modes, must be xfreed
* current_mode IN/OUT - current modes, must be xfreed */
extern void node_features_g_node_state(char **avail_modes, char **current_mode)
{
DEF_TIMERS;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
(*(ops[i].node_state))(avail_modes, current_mode);
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_node_state");
}
/* Perform set up for step launch
* mem_sort IN - Trigger sort of memory pages (KNL zonesort)
* numa_bitmap IN - NUMA nodes allocated to this job */
extern void node_features_g_step_config(bool mem_sort, bitstr_t *numa_bitmap)
{
DEF_TIMERS;
int i;
START_TIMER;
if (node_features_g_init() != SLURM_SUCCESS)
return;
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++)
(*(ops[i].step_config))(mem_sort, numa_bitmap);
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_step_config");
}
/* Test if a job's feature specification is valid */
extern int node_features_g_job_valid(char *job_features)
{
DEF_TIMERS;
int i, rc;
START_TIMER;
rc = node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && (rc == SLURM_SUCCESS)); i++)
rc = (*(ops[i].job_valid))(job_features);
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_job_valid");
return rc;
}
/* Update active and available features on specified nodes, sets features on
* all nodes is node_list is NULL */
extern int node_features_g_get_node(char *node_list)
{
DEF_TIMERS;
int i, rc;
START_TIMER;
rc = node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && (rc == SLURM_SUCCESS)); i++)
rc = (*(ops[i].get_node))(node_list);
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_get_node");
return rc;
}
/* Reset plugin configuration information */
extern int node_features_g_reconfig(void)
{
DEF_TIMERS;
int i, rc;
START_TIMER;
rc = node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && (rc == SLURM_SUCCESS)); i++)
rc = (*(ops[i].reconfig))();
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_reconfig");
return rc;
}
/* Set's the node's active features based upon job constraints.
* NOTE: Executed by the slurmd daemon.
* IN active_features - New active features
* RET error code */
extern int node_features_g_node_set(char *active_features)
{
DEF_TIMERS;
int i, rc = SLURM_SUCCESS;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && (rc == SLURM_SUCCESS)); i++) {
rc = (*(ops[i].node_set))(active_features);
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_node_set");
return rc;
}
/* Return TRUE if this (one) feature name is under this plugin's control */
extern bool node_features_g_changeable_feature(char *feature)
{
DEF_TIMERS;
int i;
bool changeable = false;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && !changeable); i++)
changeable = (*(ops[i].changeable_feature))(feature);
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_reconfig");
return changeable;
}
/* Determine if the specified user can modify the currently available node
* features */
extern bool node_features_g_user_update(uid_t uid)
{
DEF_TIMERS;
bool result = true;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; ((i < g_context_cnt) && (result == true)); i++) {
result = (*(ops[i].user_update))(uid);
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_user_update");
return result;
}
/* Return estimated reboot time, in seconds */
extern uint32_t node_features_g_boot_time(void)
{
DEF_TIMERS;
uint32_t boot_time = 0;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
boot_time = MAX(boot_time, (*(ops[i].boot_time))());
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_user_update");
return boot_time;
}
/* Return true if the plugin requires RebootProgram for booting nodes */
extern bool node_features_g_node_reboot(void)
{
DEF_TIMERS;
bool node_reboot = false;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
node_reboot = (*(ops[i].node_reboot))();
if (node_reboot)
break;
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_node_reboot");
return node_reboot;
}
/* Return bitmap of KNL nodes, NULL if none identified */
extern bitstr_t *node_features_g_get_node_bitmap(void)
{
DEF_TIMERS;
bitstr_t *node_bitmap = NULL;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
node_bitmap = (*(ops[i].get_node_bitmap))();
if (node_bitmap)
break;
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_get_node_bitmap");
return node_bitmap;
}
/*
* Return TRUE if the specified node update request is valid with respect
* to features changes (i.e. don't permit a non-KNL node to set KNL features).
*
* node_ptr IN - Pointer to struct node_record record
* update_node_msg IN - Pointer to update request
*/
extern bool node_features_g_node_update_valid(void *node_ptr,
update_node_msg_t *update_node_msg)
{
DEF_TIMERS;
bool update_valid = true;
int i;
START_TIMER;
(void) node_features_g_init();
slurm_mutex_lock(&g_context_lock);
for (i = 0; i < g_context_cnt; i++) {
update_valid = (*(ops[i].node_update_valid))(node_ptr,
update_node_msg);
if (!update_valid)
break;
}
slurm_mutex_unlock(&g_context_lock);
END_TIMER2("node_features_g_node_update_valid");
return update_valid;
}
