Jeremy Reizenstein
7f1e63aed1
Take care with single integers on gpu
...
Summary:
Pytorch seems to be becoming stricter about integer tensors of shape `(1,)` on GPU and not allowing them to be used as `int`s. For example the following no longer works on pytorch master,
foo = torch.tensor([3, 5, 3], device="cuda:0")
torch.arange(10) + foo[0]
because this is the sum of tensors on different devices.
Here fix tests which recently broke because of this.
Reviewed By: nikhilaravi
Differential Revision: D21929745
fbshipit-source-id: 25374f70468d1c895372766f1a9dd61df0833957
2020-06-10 14:13:30 -07:00
Nikhila Ravi
0eca74fa5f
lint fixes
...
Summary:
Ran the linter.
TODO: need to update the linter as per D21353065.
Reviewed By: bottler
Differential Revision: D21362270
fbshipit-source-id: ad0e781de0a29f565ad25c43bc94a19b1828c020
2020-05-04 09:56:44 -07:00
Nikhila Ravi
c3d636dc8c
Cuda updates
...
Summary:
Updates to:
- enable cuda kernel launches on any GPU (not just the default)
- cuda and contiguous checks for all kernels
- checks to ensure all tensors are on the same device
- error reporting in the cuda kernels
- cuda tests now run on a random device not just the default
Reviewed By: jcjohnson, gkioxari
Differential Revision: D21215280
fbshipit-source-id: 1bedc9fe6c35e9e920bdc4d78ed12865b1005519
2020-04-24 09:11:04 -07:00
Nikhila Ravi
b530b0af32
lint fixes
...
Summary: Resolved trailing whitespace warnings.
Reviewed By: gkioxari
Differential Revision: D21023982
fbshipit-source-id: 14ea2ca372c13cfa987acc260264ca99ce44c461
2020-04-15 21:58:59 -07:00
Georgia Gkioxari
487d4d6607
point mesh distances
...
Summary:
Implementation of point to mesh distances. The current diff contains two types:
(a) Point to Edge
(b) Point to Face
```
Benchmark Avg Time(μs) Peak Time(μs) Iterations
--------------------------------------------------------------------------------
POINT_MESH_EDGE_4_100_300_5000_cuda:0 2745 3138 183
POINT_MESH_EDGE_4_100_300_10000_cuda:0 4408 4499 114
POINT_MESH_EDGE_4_100_3000_5000_cuda:0 4978 5070 101
POINT_MESH_EDGE_4_100_3000_10000_cuda:0 9076 9187 56
POINT_MESH_EDGE_4_1000_300_5000_cuda:0 1411 1487 355
POINT_MESH_EDGE_4_1000_300_10000_cuda:0 4829 5030 104
POINT_MESH_EDGE_4_1000_3000_5000_cuda:0 7539 7620 67
POINT_MESH_EDGE_4_1000_3000_10000_cuda:0 12088 12272 42
POINT_MESH_EDGE_8_100_300_5000_cuda:0 3106 3222 161
POINT_MESH_EDGE_8_100_300_10000_cuda:0 8561 8648 59
POINT_MESH_EDGE_8_100_3000_5000_cuda:0 6932 7021 73
POINT_MESH_EDGE_8_100_3000_10000_cuda:0 24032 24176 21
POINT_MESH_EDGE_8_1000_300_5000_cuda:0 5272 5399 95
POINT_MESH_EDGE_8_1000_300_10000_cuda:0 11348 11430 45
POINT_MESH_EDGE_8_1000_3000_5000_cuda:0 17478 17683 29
POINT_MESH_EDGE_8_1000_3000_10000_cuda:0 25961 26236 20
POINT_MESH_EDGE_16_100_300_5000_cuda:0 8244 8323 61
POINT_MESH_EDGE_16_100_300_10000_cuda:0 18018 18071 28
POINT_MESH_EDGE_16_100_3000_5000_cuda:0 19428 19544 26
POINT_MESH_EDGE_16_100_3000_10000_cuda:0 44967 45135 12
POINT_MESH_EDGE_16_1000_300_5000_cuda:0 7825 7937 64
POINT_MESH_EDGE_16_1000_300_10000_cuda:0 18504 18571 28
POINT_MESH_EDGE_16_1000_3000_5000_cuda:0 65805 66132 8
POINT_MESH_EDGE_16_1000_3000_10000_cuda:0 90885 91089 6
--------------------------------------------------------------------------------
Benchmark Avg Time(μs) Peak Time(μs) Iterations
--------------------------------------------------------------------------------
POINT_MESH_FACE_4_100_300_5000_cuda:0 1561 1685 321
POINT_MESH_FACE_4_100_300_10000_cuda:0 2818 2954 178
POINT_MESH_FACE_4_100_3000_5000_cuda:0 15893 16018 32
POINT_MESH_FACE_4_100_3000_10000_cuda:0 16350 16439 31
POINT_MESH_FACE_4_1000_300_5000_cuda:0 3179 3278 158
POINT_MESH_FACE_4_1000_300_10000_cuda:0 2353 2436 213
POINT_MESH_FACE_4_1000_3000_5000_cuda:0 16262 16336 31
POINT_MESH_FACE_4_1000_3000_10000_cuda:0 9334 9448 54
POINT_MESH_FACE_8_100_300_5000_cuda:0 4377 4493 115
POINT_MESH_FACE_8_100_300_10000_cuda:0 9728 9822 52
POINT_MESH_FACE_8_100_3000_5000_cuda:0 26428 26544 19
POINT_MESH_FACE_8_100_3000_10000_cuda:0 42238 43031 12
POINT_MESH_FACE_8_1000_300_5000_cuda:0 3891 3982 129
POINT_MESH_FACE_8_1000_300_10000_cuda:0 5363 5429 94
POINT_MESH_FACE_8_1000_3000_5000_cuda:0 20998 21084 24
POINT_MESH_FACE_8_1000_3000_10000_cuda:0 39711 39897 13
POINT_MESH_FACE_16_100_300_5000_cuda:0 5955 6001 84
POINT_MESH_FACE_16_100_300_10000_cuda:0 12082 12144 42
POINT_MESH_FACE_16_100_3000_5000_cuda:0 44996 45176 12
POINT_MESH_FACE_16_100_3000_10000_cuda:0 73042 73197 7
POINT_MESH_FACE_16_1000_300_5000_cuda:0 8292 8374 61
POINT_MESH_FACE_16_1000_300_10000_cuda:0 19442 19506 26
POINT_MESH_FACE_16_1000_3000_5000_cuda:0 36059 36194 14
POINT_MESH_FACE_16_1000_3000_10000_cuda:0 64644 64822 8
--------------------------------------------------------------------------------
```
Reviewed By: jcjohnson
Differential Revision: D20590462
fbshipit-source-id: 42a39837b514a546ac9471bfaff60eefe7fae829
2020-04-11 00:21:24 -07:00
