You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
def _append_to_hooked_list(
hooked_data: dict[str, Any],
hook_name: str,
data_key: str,
element_to_append: Any,
) -> None:
"""Append the given element to a hooked list. Creates the list if it doesn't exist.
Args:
hooked_data: Dictionary of hook data that will be updated.
hook_name: Name of hook. Used as a 1st-level key in `hooked_data`.
data_key: Name of data. Used as a 2nd-level key in `hooked_data`.
element_to_append: Appended to hooked data.
"""
hooked_data.setdefault(hook_name, {}).setdefault(data_key, [])
hooked_data[hook_name][data_key].append(element_to_append)
We should convert to CPU when storing these, and handle converting back to GPU later wherever these things tensors are used.
The text was updated successfully, but these errors were encountered:
Currently,
rib.hook_fns._append_to_hooked_list
looks like:We should convert to CPU when storing these, and handle converting back to GPU later wherever these things tensors are used.
The text was updated successfully, but these errors were encountered: