Skip to content

Commit 0bed2ec

Browse files
committed
Small refinement
1 parent cc2a191 commit 0bed2ec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/ATen/native/xpu/BatchLinearAlgebra.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ void svd_kernel_xpu(
3636
auto info_cpu = empty_like_cpu(info);
3737

3838
svd_stub(
39-
A_cpu.device().type(),
39+
at::kCPU,
4040
A_cpu,
4141
full_matrices,
4242
compute_uv,

0 commit comments

Comments
 (0)