benchmark
benchmark copied to clipboard
Enable script_pdt for pytorch_struct
Summary:
Enable _script_pdt API for pytorch_struct
Test:
python run.py -m jit -d cuda pytorch_struct python run.py -m jit pytorch_struct