Searched refs:is_built_with_rocm (Results 1 – 25 of 40) sorted by relevance
12
406 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]449 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]470 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]713 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]750 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]774 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]834 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]860 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]884 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]935 if test.is_built_with_rocm():[all …]
147 if test.is_built_with_rocm():185 if test.is_built_with_rocm():
113 if test.is_built_with_rocm():183 if test.is_built_with_rocm():
36 optional_float64 = [] if test.is_built_with_rocm() else [dtypes.float64]
233 if not test.is_built_with_rocm():923 + [np.float64] if not test.is_built_with_rocm() else []:940 + [np.float64] if not test.is_built_with_rocm() else []:973 + [np.float64] if not test.is_built_with_rocm() else []:1804 if test.is_built_with_rocm():1891 if test.is_built_with_rocm():
94 def is_built_with_rocm(): function102 return is_built_with_cuda() or is_built_with_rocm()
33 self.assertEqual(build_info["is_rocm_build"], test.is_built_with_rocm())
31 test.is_built_with_rocm())
52 if test.is_built_with_rocm():137 if test.is_built_with_rocm():153 if test.is_built_with_rocm():
155 if test.is_built_with_rocm():327 if test.is_built_with_rocm():343 if test.is_built_with_rocm():372 if test.is_built_with_rocm():388 if test.is_built_with_rocm():472 if test.is_built_with_rocm():572 if test.is_built_with_rocm():754 if test.is_built_with_rocm():
258 if test.is_built_with_rocm():329 if test.is_built_with_rocm():376 if test.is_built_with_rocm():421 if test.is_built_with_rocm():597 if test.is_built_with_rocm():636 if test.is_built_with_rocm():772 if test.is_built_with_rocm():927 if test.is_built_with_rocm():
51 if test.is_built_with_rocm():147 if test.is_built_with_rocm():389 if test.is_built_with_rocm():
37 if test.is_built_with_rocm() or test_util.is_xla_enabled():
132 if not test.is_built_with_rocm():240 if not test.is_built_with_rocm():
40 name: "is_built_with_rocm"
56 name: "is_built_with_rocm"
1487 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float641498 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float641521 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float641532 if test_lib.is_built_with_rocm():1544 if test_lib.is_built_with_rocm():1572 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float641583 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float641606 dtype = np.float32 if test_lib.is_built_with_rocm() else np.float641617 if test_lib.is_built_with_rocm():1629 if test_lib.is_built_with_rocm():
73 if test.is_built_with_rocm():
87 if test.is_built_with_rocm():
435 if test.is_built_with_rocm():475 if test.is_built_with_rocm():595 if not test.is_built_with_rocm():599 if test.is_built_with_rocm():657 if test.is_built_with_rocm():792 if test.is_built_with_rocm():1184 if not test.is_built_with_rocm():
53 if test.is_built_with_rocm() and x.dtype in (np.complex64, np.complex128):67 if test.is_built_with_rocm() and x.dtype in (np.complex64, np.complex128):87 if test.is_built_with_rocm():187 if test.is_built_with_rocm():
156 if test_lib.is_built_with_rocm():
172 test.is_built_with_rocm():
95 if xla_test.test.is_built_with_rocm():
107 if not test.is_built_with_rocm():