Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "6dd3871ae05bd2a7d52c637e14db17887868aee9"
Commit fb3459ad authored by rusty1s's avatar rusty1s
Browse files

update

parent 9c3519b4
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
csrc/metis.cpp
View file @ fb3459ad
#include "cpu/metis_cpu.h"
#include <Python.h> #include <Python.h>
#include <torch/script.h> #include <torch/script.h>
#include "cpu/metis_cpu.h"
#ifdef _WIN32 #ifdef _WIN32
PyMODINIT_FUNC PyInit__metis(void) { return NULL; } PyMODINIT_FUNC PyInit__metis(void) { return NULL; }
#endif #endif
......
test/test_metis.py
View file @ fb3459ad
...@@ -12,11 +12,12 @@ def test_metis(device): ...@@ -12,11 +12,12 @@ def test_metis(device):
assert partptr.numel() == 3 assert partptr.numel() == 3
assert perm.numel() == 6 assert perm.numel() == 6
mat, partptr, perm = weighted_mat.partition(num_parts=2, recursive=True) unweighted_mat = SparseTensor.from_dense(torch.ones((6, 6), device=device))
mat, partptr, perm = unweighted_mat.partition(num_parts=2, recursive=True)
assert partptr.numel() == 3 assert partptr.numel() == 3
assert perm.numel() == 6 assert perm.numel() == 6
unweighted_mat = SparseTensor.from_dense(torch.ones((6, 6), device=device)) unweighted_mat = unweighted_mat.set_value(None)
mat, partptr, perm = unweighted_mat.partition(num_parts=2, recursive=True) mat, partptr, perm = unweighted_mat.partition(num_parts=2, recursive=True)
assert partptr.numel() == 3 assert partptr.numel() == 3
assert perm.numel() == 6 assert perm.numel() == 6
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