Rashuai/bert cuda - Githubissues

microsoft / onnxruntime-extensions

onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime

MIT License

323 stars 84 forks source link

Closed RandySheriffH closed 10 months ago

RandySheriffH commented 10 months ago

Add FastGelu as contrib op.

wejoncy commented 10 months ago

Is it possible to get the GPU device-id in custom op?

RandySheriffH commented 10 months ago

Is it possible to get the GPU device-id in custom op?

No, custom op has no way to know the device yet.