deepjavalibrary / deepjavalibrary/djl

Support for custom pytorch ops

Open
#441 7 comments 0 reactions 1 assignee Claimed by @stu1130 View on GitHub
question
Dominant language
Java
Stars
4.9k
Forks
759
Avg merge
16h 15m
Merged PRs (30d)
15

Description

## Is there a way to use torchscript model with custom ops?
I'm trying to run pytorch geometric model with DJL and have a problem with custom ops from pytorch scatter library. The error is

```
Exception in thread "main" ai.djl.engine.EngineException:
Unknown builtin op: torch_sparse::ptr2ind.
Could not find any similar ops to torch_sparse::ptr2ind. This op may not exist or may not be currently supported in TorchScript.
```

Will it be enough to just compile and install the required library or its a more complex case with JNI involved ?

Thanks in advance

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.