commit 996b6a3d00349f83086d7202dfb17e7ecad42812
author Ivan Kobzarev <ivankobzarev@fb.com> Tue May 26 11:39:19 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Tue May 26 11:42:55 2020 -0700
tree 5451f0eb08d227b3f17deea14f58e43bd5ec5da7
parent 583ff947e1f4e922ba15832c7334cdecf9cd0439

[vulkan] Fix python overrides tests for is_vulkan_available (#39016)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39016

Differential Revision: D21724619

Pulled By: IvanKobzarev

fbshipit-source-id: d7a6c8b944a55bc4f2cce957eeac08c5801667a0

torch/_overrides.py

diff --git a/torch/_overrides.py b/torch/_overrides.py
index c2542d9..cacd014 100644
--- a/torch/_overrides.py
+++ b/torch/_overrides.py
@@ -139,6 +139,7 @@
         torch.autocast_increment_nesting,
         torch.autocast_decrement_nesting,
         torch.nn.functional.hardswish,
+        torch.is_vulkan_available,
     )
 
 def get_testing_overrides():