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
12 行
378 B
Python
12 行
378 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])
|
|
|
|
from .checks import *
|
|
from .graph_cases import get_cases
|