[GH-ISSUE #801] [Bug] RuntimeError: CUDA error: invalid device function #580

Open
opened 2026-10-29 14:41:27 +00:00 by bsakaguchi · 4 comments
Owner

Originally created by @kaixwangwei on GitHub (Mar 5, 2025).
Original GitHub issue: https://github.com/kvcache-ai/ktransformers/issues/801

Originally assigned to: @Atream on GitHub.

检查清单

  • 1. 我已经搜索过相关问题,但未能获得预期的帮助
  • 2. 该问题在最新版本中尚未修复
  • 3. 请注意,如果您提交的BUG相关 issue 缺少对应环境信息和最小可复现示例,我们将难以复现和定位问题,降低获得反馈的可能性
  • 4. 如果您提出的不是bug而是问题,请在讨论区发起讨论 https://github.com/kvcache-ai/ktransformers/discussions。否则该 issue 将被关闭
  • 5. 为方便社区交流,我将使用中文/英文或附上中文/英文翻译(如使用其他语言)。未附带翻译的非中文/英语内容可能会被关闭

问题描述

运行环境:
GPU:NVIDIA GeForce GTX 1050 Ti
CPU:12th Gen Intel(R) Core(TM) i5-12500
RAM:32GB
运行命令:
python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ~/aCodes/huggleface/DeepSeek-V2-Lite-Q4_K_M-GGUF/

出现如下报错,请帮忙看看是啥原因?
Chat: who are you . for Windows or GPU before ampere, use forward_windows for Windows or GPU before ampere, use forward_windows Traceback (most recent call last): File "<frozen runpy>", line 198, in _run_module_as_main File "<frozen runpy>", line 88, in _run_code File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/local_chat.py", line 184, in <module> fire.Fire(local_chat) File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 135, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 468, in _Fire component, remaining_args = _CallAndUpdateTrace( ^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 684, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/local_chat.py", line 178, in local_chat generated = prefill_and_generate( ^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/util/utils.py", line 208, in prefill_and_generate logits = chunk_prefill(inputs[:, chunk_start:chunk_end], cache_position[chunk_start:chunk_end], past_key_values) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/util/utils.py", line 165, in chunk_prefill logits = model( ^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/models/modeling_deepseek.py", line 1731, in forward outputs = self.model( ^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/models.py", line 727, in forward layer_outputs = decoder_layer( ^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/models/modeling_deepseek.py", line 1238, in forward hidden_states, self_attn_weights, present_key_value = self.self_attn( ^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/attention.py", line 594, in forward return self.forward_windows( ^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/attention.py", line 533, in forward_windows return self.forward_chunck( ^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/attention.py", line 92, in forward_chunck compressed_kv = self.kv_a_proj_with_mqa(hidden_states) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/linear.py", line 507, in forward y = self.generate_linear.forward(x) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/linear.py", line 327, in forward padding_input[:,:self.orin_in_features] = x ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: CUDA error: invalid device function CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1 Compile with TORCH_USE_CUDA_DSAto enable device-side assertions.

复现步骤

环境步骤:https://github.com/kvcache-ai/ktransformers/blob/main/doc/en/install.md

python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ~/aCodes/huggleface/DeepSeek-V2-Lite-Q4_K_M-GGUF/

环境信息

运行环境:
GPU:NVIDIA GeForce GTX 1050 Ti
CPU:12th Gen Intel(R) Core(TM) i5-12500

Originally created by @kaixwangwei on GitHub (Mar 5, 2025). Original GitHub issue: https://github.com/kvcache-ai/ktransformers/issues/801 Originally assigned to: @Atream on GitHub. ### 检查清单 - [x] 1. 我已经搜索过相关问题,但未能获得预期的帮助 - [ ] 2. 该问题在最新版本中尚未修复 - [ ] 3. 请注意,如果您提交的BUG相关 issue 缺少对应环境信息和最小可复现示例,我们将难以复现和定位问题,降低获得反馈的可能性 - [ ] 4. 如果您提出的不是bug而是问题,请在讨论区发起讨论 https://github.com/kvcache-ai/ktransformers/discussions。否则该 issue 将被关闭 - [ ] 5. 为方便社区交流,我将使用中文/英文或附上中文/英文翻译(如使用其他语言)。未附带翻译的非中文/英语内容可能会被关闭 ### 问题描述 运行环境: GPU:NVIDIA GeForce GTX 1050 Ti CPU:12th Gen Intel(R) Core(TM) i5-12500 RAM:32GB 运行命令: python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ~/aCodes/huggleface/DeepSeek-V2-Lite-Q4_K_M-GGUF/ 出现如下报错,请帮忙看看是啥原因? `Chat: who are you . for Windows or GPU before ampere, use forward_windows for Windows or GPU before ampere, use forward_windows Traceback (most recent call last): File "<frozen runpy>", line 198, in _run_module_as_main File "<frozen runpy>", line 88, in _run_code File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/local_chat.py", line 184, in <module> fire.Fire(local_chat) File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 135, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 468, in _Fire component, remaining_args = _CallAndUpdateTrace( ^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 684, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/local_chat.py", line 178, in local_chat generated = prefill_and_generate( ^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/util/utils.py", line 208, in prefill_and_generate logits = chunk_prefill(inputs[:, chunk_start:chunk_end], cache_position[chunk_start:chunk_end], past_key_values) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/util/utils.py", line 165, in chunk_prefill logits = model( ^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/models/modeling_deepseek.py", line 1731, in forward outputs = self.model( ^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/models.py", line 727, in forward layer_outputs = decoder_layer( ^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/models/modeling_deepseek.py", line 1238, in forward hidden_states, self_attn_weights, present_key_value = self.self_attn( ^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/attention.py", line 594, in forward return self.forward_windows( ^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/attention.py", line 533, in forward_windows return self.forward_chunck( ^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/attention.py", line 92, in forward_chunck compressed_kv = self.kv_a_proj_with_mqa(hidden_states) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/linear.py", line 507, in forward y = self.generate_linear.forward(x) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/wangwei/miniconda3/envs/ktransformers/lib/python3.11/site-packages/ktransformers/operators/linear.py", line 327, in forward padding_input[:,:self.orin_in_features] = x ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: CUDA error: invalid device function CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1 Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions. ` ### 复现步骤 环境步骤:https://github.com/kvcache-ai/ktransformers/blob/main/doc/en/install.md python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ~/aCodes/huggleface/DeepSeek-V2-Lite-Q4_K_M-GGUF/ ### 环境信息 运行环境: GPU:NVIDIA GeForce GTX 1050 Ti CPU:12th Gen Intel(R) Core(TM) i5-12500
Author
Owner

@loverara commented on GitHub (Mar 5, 2025):

一样的问题

<!-- gh-comment-id:2700628252 --> @loverara commented on GitHub (Mar 5, 2025): 一样的问题
Author
Owner

@kosehy commented on GitHub (Mar 8, 2025):

When I use Quadro P5000, I also face same issue

python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ./DeepSeek-V2-Lite-Chat-GGUF

Chat: 
for Windows or GPU before ampere, use forward_windows
for Windows or GPU before ampere, use forward_windows
Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/new-pool/ktransformers/ktransformers/local_chat.py", line 184, in <module>
    fire.Fire(local_chat)
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 135, in Fire
    component_trace = _Fire(component, args, parsed_flag_args, context, name)
                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 468, in _Fire
    component, remaining_args = _CallAndUpdateTrace(
                                ^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 684, in _CallAndUpdateTrace
    component = fn(*varargs, **kwargs)
                ^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/local_chat.py", line 178, in local_chat
    generated = prefill_and_generate(
                ^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/util/utils.py", line 208, in prefill_and_generate
    logits = chunk_prefill(inputs[:, chunk_start:chunk_end], cache_position[chunk_start:chunk_end], past_key_values)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/util/utils.py", line 165, in chunk_prefill
    logits = model(
             ^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1731, in forward
    outputs = self.model(
              ^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/operators/models.py", line 727, in forward
    layer_outputs = decoder_layer(
                    ^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1238, in forward
    hidden_states, self_attn_weights, present_key_value = self.self_attn(
                                                          ^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 594, in forward
    return self.forward_windows(
           ^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 533, in forward_windows
    return self.forward_chunck(
           ^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 92, in forward_chunck
    compressed_kv = self.kv_a_proj_with_mqa(hidden_states)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 507, in forward
    y = self.generate_linear.forward(x)
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 327, in forward
    padding_input[:,:self.orin_in_features] = x
    ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: CUDA error: invalid device function
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
<!-- gh-comment-id:2708222861 --> @kosehy commented on GitHub (Mar 8, 2025): When I use `Quadro P5000`, I also face same issue ```shell python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ./DeepSeek-V2-Lite-Chat-GGUF Chat: for Windows or GPU before ampere, use forward_windows for Windows or GPU before ampere, use forward_windows Traceback (most recent call last): File "<frozen runpy>", line 198, in _run_module_as_main File "<frozen runpy>", line 88, in _run_code File "/new-pool/ktransformers/ktransformers/local_chat.py", line 184, in <module> fire.Fire(local_chat) File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 135, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 468, in _Fire component, remaining_args = _CallAndUpdateTrace( ^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 684, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/local_chat.py", line 178, in local_chat generated = prefill_and_generate( ^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/util/utils.py", line 208, in prefill_and_generate logits = chunk_prefill(inputs[:, chunk_start:chunk_end], cache_position[chunk_start:chunk_end], past_key_values) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/util/utils.py", line 165, in chunk_prefill logits = model( ^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1731, in forward outputs = self.model( ^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/operators/models.py", line 727, in forward layer_outputs = decoder_layer( ^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1238, in forward hidden_states, self_attn_weights, present_key_value = self.self_attn( ^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 594, in forward return self.forward_windows( ^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 533, in forward_windows return self.forward_chunck( ^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 92, in forward_chunck compressed_kv = self.kv_a_proj_with_mqa(hidden_states) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 507, in forward y = self.generate_linear.forward(x) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 327, in forward padding_input[:,:self.orin_in_features] = x ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: CUDA error: invalid device function Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions. ```
Author
Owner

@kosehy commented on GitHub (Mar 8, 2025):

When I use Quadro P5000, I also face same issue

python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ./DeepSeek-V2-Lite-Chat-GGUF

Chat:
for Windows or GPU before ampere, use forward_windows
for Windows or GPU before ampere, use forward_windows
Traceback (most recent call last):
File "", line 198, in _run_module_as_main
File "", line 88, in _run_code
File "/new-pool/ktransformers/ktransformers/local_chat.py", line 184, in
fire.Fire(local_chat)
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 135, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 468, in _Fire
component, remaining_args = _CallAndUpdateTrace(
^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 684, in _CallAndUpdateTrace
component = fn(*varargs, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/local_chat.py", line 178, in local_chat
generated = prefill_and_generate(
^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/util/utils.py", line 208, in prefill_and_generate
logits = chunk_prefill(inputs[:, chunk_start:chunk_end], cache_position[chunk_start:chunk_end], past_key_values)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/util/utils.py", line 165, in chunk_prefill
logits = model(
^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1731, in forward
outputs = self.model(
^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/operators/models.py", line 727, in forward
layer_outputs = decoder_layer(
^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1238, in forward
hidden_states, self_attn_weights, present_key_value = self.self_attn(
^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 594, in forward
return self.forward_windows(
^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 533, in forward_windows
return self.forward_chunck(
^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 92, in forward_chunck
compressed_kv = self.kv_a_proj_with_mqa(hidden_states)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 507, in forward
y = self.generate_linear.forward(x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 327, in forward
padding_input[:,:self.orin_in_features] = x
~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: CUDA error: invalid device function
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

I followed #374 and Pyjacc's guideline in #425 and finally run the demo code using P5000 pascal architecture.

<!-- gh-comment-id:2708241556 --> @kosehy commented on GitHub (Mar 8, 2025): > When I use `Quadro P5000`, I also face same issue > > python -m ktransformers.local_chat --model_path deepseek-ai/DeepSeek-V2-Lite-Chat --gguf_path ./DeepSeek-V2-Lite-Chat-GGUF > > Chat: > for Windows or GPU before ampere, use forward_windows > for Windows or GPU before ampere, use forward_windows > Traceback (most recent call last): > File "<frozen runpy>", line 198, in _run_module_as_main > File "<frozen runpy>", line 88, in _run_code > File "/new-pool/ktransformers/ktransformers/local_chat.py", line 184, in <module> > fire.Fire(local_chat) > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 135, in Fire > component_trace = _Fire(component, args, parsed_flag_args, context, name) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 468, in _Fire > component, remaining_args = _CallAndUpdateTrace( > ^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/fire/core.py", line 684, in _CallAndUpdateTrace > component = fn(*varargs, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/local_chat.py", line 178, in local_chat > generated = prefill_and_generate( > ^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/util/utils.py", line 208, in prefill_and_generate > logits = chunk_prefill(inputs[:, chunk_start:chunk_end], cache_position[chunk_start:chunk_end], past_key_values) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/util/utils.py", line 165, in chunk_prefill > logits = model( > ^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl > return self._call_impl(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl > return forward_call(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1731, in forward > outputs = self.model( > ^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl > return self._call_impl(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl > return forward_call(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/operators/models.py", line 727, in forward > layer_outputs = decoder_layer( > ^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl > return self._call_impl(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl > return forward_call(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/models/modeling_deepseek.py", line 1238, in forward > hidden_states, self_attn_weights, present_key_value = self.self_attn( > ^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl > return self._call_impl(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl > return forward_call(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 594, in forward > return self.forward_windows( > ^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 533, in forward_windows > return self.forward_chunck( > ^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/operators/attention.py", line 92, in forward_chunck > compressed_kv = self.kv_a_proj_with_mqa(hidden_states) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl > return self._call_impl(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/miniconda3/envs/ktransformers/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl > return forward_call(*args, **kwargs) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 507, in forward > y = self.generate_linear.forward(x) > ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ > File "/new-pool/ktransformers/ktransformers/operators/linear.py", line 327, in forward > padding_input[:,:self.orin_in_features] = x > ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^ > RuntimeError: CUDA error: invalid device function > Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions. I followed #374 and Pyjacc's guideline in #425 and finally run the demo code using P5000 pascal architecture.
Author
Owner

@save-FGG commented on GitHub (Mar 12, 2025):

我也遇到了,换了模型,改了运行参数还是一问问题就提示这个

<!-- gh-comment-id:2717178257 --> @save-FGG commented on GitHub (Mar 12, 2025): 我也遇到了,换了模型,改了运行参数还是一问问题就提示这个
Sign in to join this conversation.
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
kvcache-ai/ktransformers#580
No description provided.