search

OpenGVLab / video-mamba-suite

The suite of modeling video with Mamba
MIT License
216 stars 21 forks source link

About causal_conv1d_cuda library #15

Open Xingguang-Li opened 3 months ago

Xingguang-Li commented 3 months ago

Thank you for your excellent work.

I'm going to read your mamba block carefully, I'm just going to learn about your structure, so I didn't install the environment you provided, and you also provided the mamba file, but while reading, I had a difficulty:

conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd(x, conv1d_weight, conv1d_bias, True).

When I read this line of code, I tried hard to find causal_conv1d_cuda library, I wanted to understand the contents of it, but I couldn't find it, can you help me find it, or give me some pointers and tell me what to do.

cg1177 commented 3 months ago

causal conv1 is installed as python package. The code is here

Xingguang-Li commented 3 months ago

After so much exploration, I still haven't solved my problem, and allow me to rephrase it. I'm trying to visualize the functions in the selective_scan_cuda.fwd library to explore the network model of the DBM blocks you proposed, but I can't find a way to find a way to do it, and I can't read the cuda file that ends in cu. Please give me some advice or point out my mistakes (try to be clear) Thank you very much for your help!