nll libtorch

[ACSadighi]

Still needs debugging. Not sure what I'm doing wrong.

I map the string argument to integers in NDArray.chpl and unmap in bridge.cpp.

[Iainmon]

What error are you getting when trying to compile this?

[ACSadighi]

python3 correspondence.py --test-only loss/nll --print-compiler-errors
Traceback (most recent call last):
File "/nfs/stak/users/sadighia/.local/lib/python3.9/site-packages/torch/init.py", line 318, in _load_global_deps
ctypes.CDLL(global_deps_lib_path, mode=ctypes.RTLD_GLOBAL)
File "/usr/lib64/python3.9/ctypes/init.py", line 374, in init
self._handle = _dlopen(self._name, mode)
OSError: libcudart.so.12: cannot open shared object file: No such file or directory

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/nfs/stak/users/sadighia/capstone/ChAI/test/correspondence/correspondence.py", line 4, in
import torch
File "/nfs/stak/users/sadighia/.local/lib/python3.9/site-packages/torch/init.py", line 404, in
_load_global_deps()
File "/nfs/stak/users/sadighia/.local/lib/python3.9/site-packages/torch/init.py", line 362, in _load_global_deps
_preload_cuda_deps(lib_folder, lib_name)
File "/nfs/stak/users/sadighia/.local/lib/python3.9/site-packages/torch/init.py", line 302, in _preload_cuda_deps
raise ValueError(f"{lib_name} not found in the system path {sys.path}")
ValueError: libcublas.so.*[0-9] not found in the system path ['/nfs/stak/users/sadighia/capstone/ChAI/test/correspondence', '/usr/lib64/python39.zip', '/usr/lib64/python3.9', '/usr/lib64/python3.9/lib-dynload', '/nfs/stak/users/sadighia/.local/lib/python3.9/site-packages', '/usr/local/lib64/python3.9/site-packages', '/usr/local/lib/python3.9/site-packages', '/usr/lib64/python3.9/site-packages', '/usr/lib/python3.9/site-packages']

[ACSadighi]

NLL function passes tests. The changes we made with commenting things out yesterday probably makes a merge difficult, but it works on my branch.