I tried to compile this CUDA extension in MONAI docker directly, it completed successfully.
And tried to load and test:

>>> import torch
>>> import lltm_cuda
>>> lltm_cuda.forward
<built-in method forward of PyCapsule object at 0x7fc9061fc090>

Thanks.

/black

/black

some mypy issue here?

some mypy issue here?

Yes, now I updated according to the latest tutorial code.
Thanks.

I feel this feature is an important and unique one, shall we make it a top-level module, such as monai.ops.lltm_cuda ("ops" short for "operations"), and in the monai.networks.LLTM module, we optionally import monai.ops.lltm_cuda.

could you help remove this temporary file, I think it is included by mistake: https://github.com/Project-MONAI/MONAI/blob/master/tests/testing_data/._Task04_Hippocampus

could you help remove this temporary file, I think it is included by mistake: https://github.com/Project-MONAI/MONAI/blob/master/tests/testing_data/._Task04_Hippocampus

Sure, will remove it in the next commit.
Thanks.

/black

@Nic-Ma progress so far:

1. in our case, pip will use pyproject.toml to configure the building process but our pyproject.toml doesn't look good, (missing a [build-system] section)

to resolve 1:
option a, remove pyproject.toml <- can't do it because psf/black needs it
option b, we can add a build-system section <- this causes further issues about the inconsistent torch version in build (always the latest) and run (we test different torch versions) the cpp extensions

looks like we need to address option b properly

additionally, there's a mypy issue...

Hi @wyli ,

Thanks very much for your help!

Hi @wyli ,

Thanks very much for your help here, I think it looks good to me now.

this is ready @Nic-Ma (delayed because I wanted to add some pip install cache so that the CI tests are slightly more robust)

this is ready @Nic-Ma (delayed because I wanted to add some pip install cache so that the CI tests are slightly more robust)

OK, sounds good!
Thanks.