i try to use torch_tensorrt.compiler() with a model and i got that
File “/home/cplus/projects/m.tarek_master/Image_enhancement/UW_CycleGAN/swin2sr/load_ttrt_model.py”, line 148, in
trt_model_2 = compile_with_torch_tensorrt(model=model.eval(),
File “/home/cplus/projects/m.tarek_master/Image_enhancement/UW_CycleGAN/swin2sr/load_ttrt_model.py”, line 117, in compile_with_torch_tensorrt
trt_model = torch_tensorrt.compile(
File “/home/cplus/projects/m.tarek_master/python_enviornments/virtual_env/lib/python3.10/site-packages/torch_tensorrt/_compile.py”, line 232, in compile
compiled_ts_module: torch.jit.ScriptModule = torchscript_compile(
File “/home/cplus/projects/m.tarek_master/python_enviornments/virtual_env/lib/python3.10/site-packages/torch_tensorrt/ts/_compiler.py”, line 161, in compile
compiled_cpp_mod = _C.compile_graph(module._c, _parse_compile_spec(spec))
RuntimeError: 0 INTERNAL ASSERT FAILED at “/pytorch/torch/csrc/jit/ir/alias_analysis.cpp”:616, please report a bug to PyTorch. We don’t have an op for aten::floor_divide but it isn’t a special case. Argument types: int, int,
Candidates:
aten::floor_divide(Tensor self, Tensor other) → Tensor
aten::floor_divide.Scalar(Tensor self, Scalar other) → Tensor
aten::floor_divide.out(Tensor self, Tensor other, *, Tensor(a!) out) → Tensor(a!)
aten::floor_divide.Scalar_out(Tensor self, Scalar other, *, Tensor(a!) out) → Tensor(a!)
Process finished with exit code 1