Not able to run the python tests

[sheetalarkadam]

I am not able to run the benchmark.py, float_test.py and ans_test.py. The line torch.ops.load_library("//dietgpu:dietgpu") fails with file not found. I can see there is an open PR that addresses this issue by changing the path. But I still get undefined symbol error.

Any steps to test the dietgpu performance and use the test scripts would greatly help