Skip to content

Commit 1d1f352

Browse files
authored
Run test ops tests from outside of pytorch root folder (#1676)
1 parent 53b5b02 commit 1d1f352

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.github/scripts/validate_test_ops.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ pushd pytorch
2020

2121
pip install expecttest numpy pyyaml jinja2 packaging hypothesis unittest-xml-reporting scipy
2222

23-
# Run pytorch cuda wheels validation
23+
# Run pytorch cuda wheels validation
2424
# Detect ReduceLogicKernel (ReduceOp and kernel) IMA
2525
python test/test_ops.py -k test_dtypes_all_cuda
2626
# Detect BinaryMulKernel (elementwise binary functor internal mul) IMA
@@ -29,7 +29,10 @@ python test/test_torch.py -k test_index_reduce_reduce_prod_cuda_int32
2929
python test/test_binary_ufuncs.py -k test_contig_vs_every_other___rand___cuda_int32
3030
# Detect MaxMinElementwiseKernel (maximum) IMA
3131
python test/test_schema_check.py -k test_schema_correctness_clamp_cuda_int8
32+
33+
pushd /tmp
3234
# Detect StepKernel (nextafter) IMA
3335
python -c "import torch; print(torch.nextafter(torch.tensor([-4.5149, -5.9053, -0.9516, -2.3615, 1.5591], device='cuda:0'), torch.tensor(3.8075, device='cuda:0')))"
3436
# Detect BinaryGeometricKernels (atan2) IMA
3537
python -c "import torch; x = (torch.randn((2,1,1), dtype=torch.float, device='cuda')*5).to(torch.float32); y=(torch.randn((), dtype=torch.float, device='cuda')*5).to(torch.float32); print(torch.atan2(x,y))"
38+
popd

0 commit comments

Comments
 (0)