Jart synapse

[ZhenmingYu]

Related issues

Support a new device type based on JART V1b memristor model and test scripts for that.
Details can be found at https://ieeexplore.ieee.org/document/10052010

Fixing the issues in #583 and #679 about building errors

Description

Deleted CUDA related codes, fixed JART CPU version.
Still not able to run properly, issues relates to device.py and the linkage between python and c++ code.

Details

My original implementation was with the BUILD_PULSED_DEVICE_CONSTRUCTORS (commented out on line 540).
@maljoras changed it to the current implementation because he thinks some parts are not necessary for my device as I remembered.

I don't really have a good understanding of this section about managing the parameters and how they go in and out of the memory, so I'm not comfortable about changing stuff here.

However, by comparing with the BUILD_PULSED_DEVICE_CONSTRUCTORS, it seems that a "public:" was missing at line 266?
And adding that seems to have solved the compilation issue.

I don't really know the code here is correct or whether there are any problems.
But anyhow the examples are still not running yet, which means some part between the python-c++ linkage is still not correct.

My current error output:
(.venv) zyu@pgi15-gpu2:~/aihwkit$ python examples/03_mnist_training.py
Traceback (most recent call last):
File "/Users/zyu/aihwkit/examples/03_mnist_training.py", line 27, in
from aihwkit.nn import AnalogLinear, AnalogSequential
File "/Users/zyu/aihwkit/src/aihwkit/nn/init.py", line 11, in
from aihwkit.nn.modules.container import AnalogSequential, AnalogWrapper
File "/Users/zyu/aihwkit/src/aihwkit/nn/modules/container.py", line 17, in
from aihwkit.nn.modules.base import AnalogLayerBase
File "/Users/zyu/aihwkit/src/aihwkit/nn/modules/base.py", line 16, in
from aihwkit.simulator.tiles.module import TileModule
File "/Users/zyu/aihwkit/src/aihwkit/simulator/tiles/init.py", line 11, in
from aihwkit.simulator.tiles.analog import AnalogTile
File "/Users/zyu/aihwkit/src/aihwkit/simulator/tiles/analog.py", line 15, in
from aihwkit.simulator.tiles.rpucuda import RPUCudaSimulatorTileWrapper
File "/Users/zyu/aihwkit/src/aihwkit/simulator/tiles/rpucuda.py", line 18, in
from aihwkit.simulator.parameters.base import RPUConfigGeneric
File "/Users/zyu/aihwkit/src/aihwkit/simulator/parameters/init.py", line 25, in
from .training import UpdateParameters
File "/Users/zyu/aihwkit/src/aihwkit/simulator/parameters/training.py", line 14, in
from .helpers import _PrintableMixin
File "/Users/zyu/aihwkit/src/aihwkit/simulator/parameters/helpers.py", line 14, in
from aihwkit.simulator import rpu_base
ImportError: /Users/zyu/aihwkit/src/aihwkit/simulator/rpu_base.cpython-312-x86_64-linux-gnu.so: undefined symbol: _ZN3RPU29JARTv1bRPUDeviceMetaParameterIfE10initializeEv

[coreylammie]

Hi @ZhenmingYu,

It looks like you have a linking error and the following de-mangled symbol is not being found: RPU::JARTv1bRPUDeviceMetaParameter<float>::initialize(). I would suggest adding a new line in CMakeLists here: https://github.com/IBM/aihwkit/blob/master/CMakeLists.txt#L165, message(${EXTENSION_BINDINGS_SRCS}) to check whether it is being correctly associated.