def test_connection_type_on_demand(self): """Connection type ON_DEMAND""" result = run_connection_type("ON_DEMAND") # Assert correct result. assert_result_equals_tensor_value( result['result'], np.array([3.0, 8.0], dtype=np.float32)) # Assert single occurences of attach and compile # with compilation occuring first. assert (len(result['attach']) == 1), "Missing attach" assert (len(result['compile']) == 1), "Missing compile" assert (result['attach'][0] > result['compile'][0]), "Compile after attach"
def test_connection_type_always(self): """Connection type ALWAYS""" result = run_connection_type("ALWAYS") # Assert correct result. assert_result_equals_tensor_value( result['result'], np.array([3.0, 8.0], dtype=np.float32)) # Assert single occurences of attach and compile # with attach occuring first. assert (len(result['attach']) == 1), "Missing attach" assert (len(result['compile']) == 1), "Missing compile" assert (result['attach'][0] < result['compile'][0]), "Compile before attach"
def test_auto_sharding(self): """Automatic sharding example using 2 shards""" out = run_simple_sharding(True) assert_result_equals_tensor_value( out, np.array([3.0, 8.0], dtype=np.float32))
def test_manual_sharding(self): """Manual sharding example using 2 shards""" out = run_simple_sharding(False) assert_result_equals_tensor_value( out, np.array([3.0, 8.0], dtype=np.float32))