dmlc--dgl
Minjie Wang
f4608c2227
* COO sort * COOToCSR * CSR2COO * CSRSort; CSRTranspose * pass all CSR tests * lint * remove int32 conversion * fix tensorflow nn tests * turn on CI * fix * addreess comments
16 行
448 B
Python
16 行
448 B
Python
import pytest
|
|
import backend as F
|
|
|
|
if F._default_context_str == 'cpu':
|
|
parametrize_dtype = pytest.mark.parametrize("idtype", [F.int32, F.int64])
|
|
else:
|
|
# only test int32 on GPU because many graph operators are not supported for int64.
|
|
parametrize_dtype = pytest.mark.parametrize("idtype", [F.int32, F.int64])
|
|
|
|
def check_fail(fn, *args, **kwargs):
|
|
try:
|
|
fn(*args, **kwargs)
|
|
return False
|
|
except:
|
|
return True
|