Remove deprecated FP16 tests

test/test_data_type.py

@@ -14,8 +14,6 @@ def setUp(cls):
     cls.original_env = {
         'XLA_USE_BF16': os.environ.get('XLA_USE_BF16'),
         'XLA_DOWNCAST_BF16': os.environ.get('XLA_DOWNCAST_BF16'),
-        'XLA_USE_FP16': os.environ.get('XLA_USE_FP16'),
-        'XLA_DOWNCAST_FP16': os.environ.get('XLA_DOWNCAST_FP16'),
         'XLA_USE_32BIT_LONG': os.environ.get('XLA_USE_32BIT_LONG')
     }
 
@@ -46,21 +44,11 @@ def test_datatype_use_bf16(self):
     self._test_datatype(torch.double, 'bf16', torch.floor_divide)
     self._test_datatype(torch.float, 'bf16', torch.floor_divide)
 
-  def test_datatype_use_fp16(self):
-    self._set_env(XLA_USE_FP16='1')
-    self._test_datatype(torch.double, 'bf16', torch.floor_divide)
-    self._test_datatype(torch.float, 'f16', torch.floor_divide)
-
   def test_datatype_downcast_bf16(self):
     self._set_env(XLA_DOWNCAST_BF16='1')
     self._test_datatype(torch.double, 'bf16', torch.floor_divide)
     self._test_datatype(torch.float, 'bf16', torch.floor_divide)
 
-  def test_datatype_downcast_fp16(self):
-    self._set_env(XLA_DOWNCAST_FP16='1')
-    self._test_datatype(torch.double, 'f16', torch.floor_divide)
-    self._test_datatype(torch.float, 'f16', torch.floor_divide)
-
   def test_datatype_use_32bit_long(self):
     self._set_env(XLA_USE_32BIT_LONG='1')
     self._test_datatype(torch.int64, 's32', torch.add)
@@ -86,9 +74,7 @@ def test_module_to_dtype(self):
 if __name__ == '__main__':
   suite = unittest.TestSuite()
   suite.addTest(XlaDataTypeTest("test_datatype_use_bf16"))
-  suite.addTest(XlaDataTypeTest("test_datatype_use_fp16"))
   suite.addTest(XlaDataTypeTest("test_datatype_downcast_bf16"))
-  suite.addTest(XlaDataTypeTest("test_datatype_downcast_fp16"))
   suite.addTest(XlaDataTypeTest("test_datatype_use_32bit_long"))
   suite.addTest(XlaDataTypeTest("test_module_to_dtype"))
   runner = unittest.TextTestRunner(failfast=True)