Skip to content

issue/1129 - fix dtype generalization and used variables #1130

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
wooway777 merged 1 commit into from
Apr 8, 2026
Merged

issue/1129 - fix dtype generalization and used variables #1130

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/infiniop/ops/huber_loss/nvidia/huber_loss_nvidia.cu
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ infiniStatus_t Descriptor::calculate(
launch_kernel<half>(output, input, target, workspace, _info, stream);
break;
case INFINI_DTYPE_BF16:
launch_kernel<nv_bfloat16>(output, input, target, workspace, _info, stream);
launch_kernel<cuda_bfloat16>(output, input, target, workspace, _info, stream);
break;
case INFINI_DTYPE_F32:
launch_kernel<float>(output, input, target, workspace, _info, stream);
Expand Down
2 changes: 1 addition & 1 deletion src/infiniop/ops/kthvalue/cuda/kernel.cuh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ __global__ void kthvalue_kernel(
for (unsigned int size = 2; size <= power_of_2_dim; size <<= 1) {
// Bitonic Merge
// dir: 升序或降序交替，构造双调序列
bool dir = (tid & (size / 2)) == 0;
// bool dir = (tid & (size / 2)) == 0;

// 这里的逻辑稍微复杂，为了简单和稳定，我们使用全升序排序逻辑
// 标准 Bitonic Sort 代码如下：
Expand Down
Loading