search

state-spaces / mamba

Mamba SSM architecture
Apache License 2.0
12.55k stars 1.05k forks source link

where is the src.ops.triton.k_activations? #349

Open gaotai8 opened 3 months ago

gaotai8 commented 3 months ago

import torch from mamba_ssm import Mamba Traceback (most recent call last): File "", line 1, in File "/home/gaotai/project/mamba/mamba_ssm/init.py", line 5, in from mamba_ssm.modules.mamba2 import Mamba2 File "/home/gaotai/project/mamba/mamba_ssm/modules/mamba2.py", line 26, in from mamba_ssm.ops.triton.ssd_combined import mamba_chunk_scan_combined File "/home/gaotai/project/mamba/mamba_ssm/ops/triton/ssd_combined.py", line 40, in from src.ops.triton.k_activations import _swiglu_fwd, _swiglu_bwd ModuleNotFoundError: No module named 'src.ops.triton.k_activations'

jpilaul commented 3 months ago

Should be here: https://github.com/state-spaces/mamba/blob/main/mamba_ssm/ops/triton/k_activations.py

gaotai8 commented 3 months ago

应该在这里：https://github.com/state-spaces/mamba/blob/main/mamba_ssm/ops/triton/k_activations.py

thank you.we can download v2.3.0