help wantedmodule: c++ frontendmodule: opstorchscript
描述
🐛 Bug
I'm testing about jit in C++ ( libtorch )
My test cases
int main(int argc, const char* argv[]) {
auto& ops = torch::jit::getAllOperators();
std::cout << "torch jit operators\n";
for (auto& op: ops) {
auto& name = op->schema().name();
if (name.find("torchvision") != std::string::npos)
std::cout << "op : " << op->schema().name() << "\n";
}
std::cout << "\n";
return 0;
}
>>> torch jit operators
Expected (I changed some codes in vision)
>>> torch jit operators
op : torchvision::_cuda_version
op : torchvision::deform_conv2d
op : torchvision::ps_roi_pool
op : torchvision::ps_roi_align
op : torchvision::_new_empty_tensor_op
op : torchvision::roi_pool
op : torchvision::roi_align
op : torchvision::nms
In 1st case, seems no jit opeators registry on libtorch.
Is this only occurs my env? or some codes are wrong?
Because I checked expected output when I changed srcs.
Environment
Install command follows on reamdme
torch 1.5.0 and torchvision 0.6.0