How to create node class for a tree structure with jitclass

stuartarchibald · July 21, 2020, 10:36am

Perhaps this example for a linked list using jitclass will help:

numba/numba-examples/blob/master/legacy/linkedlist.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
This example demonstrates jitclasses and deferred types for writing a
singly-linked-list.
"""
from __future__ import print_function, absolute_import
from collections import OrderedDict
import numpy as np
from numba import njit
from numba import int32, deferred_type, optional
from numba.core.runtime import rtsys
from numba.experimental import jitclass

node_type = deferred_type()

spec = OrderedDict()
spec['data'] = int32
spec['next'] = optional(node_type)

This file has been truncated. show original

Topic		Replies	Views
Jitclass with a deferred type inside a typed list used as a parameter for another jitclass Support: What is this error message?	1	165	November 8, 2023
Recursive python type annotations in @jitclass Numba	2	725	February 16, 2021
Nest dict with jit class Community Support	1	1846	September 2, 2020
How do I create a jitclass that takes a list of jitclass objects? Support: How do I do ...?	1	2608	November 29, 2020
List of arrays in jitclass Support: How do I do ...?	1	405	June 10, 2022

How to create node class for a tree structure with jitclass

Related Topics