File tree 1 file changed +11
-1
lines changed
onnxscript/tests/function_libs/torch_lib 1 file changed +11
-1
lines changed Original file line number Diff line number Diff line change 1
- # Test op correctness by comparing with PyTorch results.
1
+ # Test op correctness by comparing with PyTorch results
2
2
3
3
## Usage
4
4
@@ -13,6 +13,16 @@ pytest onnxscript/tests/function_libs/torch_lib/ops_test.py -k ceil
13
13
pytest onnxscript/tests/function_libs/torch_lib/ops_test.py -k nn_functional_scaled_dot_product_attention
14
14
```
15
15
16
+ ### Environment variables
17
+
18
+ 1 . Set environment variable ` CATCH_ORT_SEGFAULT=1 ` to catch segmentation faults
19
+ in onnxruntime by running the inference sessions in a separate process.
20
+ 2 . Set ` CREATE_REPRODUCTION_REPORT=1 ` to create markdown files for reproduction of errors. E.g.
21
+
22
+ ``` bash
23
+ CREATE_REPRODUCTION_REPORT=1 python -m pytest onnxscript/tests/function_libs/torch_lib/ops_test.py -k div_mode_int
24
+ ```
25
+
16
26
# # How to add a new operator test
17
27
18
28
See _usage_ in [ops_test_data.py](./ops_test_data.py)
You can’t perform that action at this time.
0 commit comments