hi, there is a bug when running almost finish

VoVAllen / tf-dlpack

DLPack for Tensorflow

Apache License 2.0

36 stars 4 forks source link

Closed ersanliqiao closed 3 years ago

ersanliqiao commented 3 years ago

tf2.3

VoVAllen commented 3 years ago

DLPack is officially supported in tf 2.3 by tf.experimental.dlpack.from_dlpack/to_dlpack. You no longer need this repo. Are you using dgl?

ersanliqiao commented 3 years ago

OK, i will try it then. what is dgl?

VoVAllen commented 3 years ago

DGL is another project using this repo before

ersanliqiao commented 3 years ago

Do you have any example to convert from numba to tensorflow tensor by tf2.3?

VoVAllen commented 3 years ago

@ersanliqiao I think numba just use cupy right? Cupy also supports dlpack. Did you meet any problem when using that?

ersanliqiao commented 3 years ago

also

OK，I have done it correctly, thank you very much.