Skip to content

GitLab

"...text-generation-inference.git" did not exist on "cd298bc5e5f1beb300c05099eb8d13460f867863"
Unverified Commit 2ebfa6ea authored by xiangyuzhi's avatar xiangyuzhi Committed by GitHub
Browse files

[Sparse]Fix device check (#6409) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-24-117.ap-northeast-1.compute.internal>
parent 617979d6
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
dgl_sparse/src/sddmm.cc
View file @ 2ebfa6ea
...@@ -61,8 +61,9 @@ void _SDDMMSanityCheck( ...@@ -61,8 +61,9 @@ void _SDDMMSanityCheck(
mat1.dtype() == mat2.dtype(), mat1.dtype() == mat2.dtype(),
"SDDMM: the two dense matrices should have the same dtype."); "SDDMM: the two dense matrices should have the same dtype.");
TORCH_CHECK( TORCH_CHECK(
mat1.device() == mat2.device(), mat1.device() == mat2.device() && sparse_mat->device() == mat2.device(),
"SDDMM: the two dense matrices should on the same device."); "SDDMM: the two dense matrices and sparse matrix should on the same "
"device.");
} }
torch::Tensor SDDMMAutoGrad::forward( torch::Tensor SDDMMAutoGrad::forward(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment