Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[22027] Update HW Constraints node to receive data from extra_data field #4

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

[22027] Update HW Constraints node to receive data from extra_data field #4

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 25 additions & 2 deletions hw_constraints_node.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,9 @@
from sustainml_py.nodes.HardwareConstraintsNode import HardwareConstraintsNode

# Manage signaling
import ctypes
import json
import numpy as np
import signal
import threading
import time
Expand All @@ -35,9 +38,29 @@ def signal_handler(sig, frame):
# Outputs: node_status, hw_constraints
def task_callback(user_input, node_status, hw_constraints):

# Callback implementation here
# Default values
hw_req = "PIM_AI_1chip"
mem_footprint = 100

hw_constraints.max_memory_footprint(100)
# Check if extra data has been sent
if user_input.extra_data().size() != 0:
buffer = ctypes.c_ubyte * user_input.extra_data().size()
buffer = buffer.from_address(int(user_input.extra_data().get_buffer()))
extra_data = np.frombuffer(buffer, dtype=np.uint8)
extra_data_str = extra_data.tobytes().decode('utf-8', errors='ignore')
try:
json_obj = json.loads(extra_data_str)
if json_obj is not None:
mem_footprint = int(json_obj["max_memory_footprint"])
hw_req = json_obj["hardware_required"]
except:
print("Extra data is not a valid JSON object, using default values")

# TODO parse other possible data hidden in the extra_data field, if any
# TODO populate the hw_constraints object with the required data

hw_constraints.max_memory_footprint(mem_footprint)
hw_constraints.hardware_required([hw_req])

# Main workflow routine
def run():
Expand Down