TensorFlow.NET/test/TensorFlowNET.UnitTest/NativeAPI/Eager/CApi.Eager.Context.cs

using Microsoft.VisualStudio.TestTools.UnitTesting;
using System;
using Tensorflow;

namespace TensorFlowNET.UnitTest.NativeAPI
{
    public partial class CApiEagerTest
    {
        /// <summary>
        /// TEST(CAPI, Context)
        /// </summary>
        [TestMethod]
        public void Context()
        {
            using var status = c_api.TF_NewStatus();
            var opts = c_api.TFE_NewContextOptions();

            IntPtr devices;
            using (var ctx = c_api.TFE_NewContext(opts, status))
            {
                c_api.TFE_DeleteContextOptions(opts);

                devices = c_api.TFE_ContextListDevices(ctx, status);
                EXPECT_EQ(TF_OK, TF_GetCode(status), TF_Message(status));
            }

            EXPECT_EQ(TF_OK, TF_GetCode(status), TF_Message(status));

            int num_devices = c_api.TF_DeviceListCount(devices);
            EXPECT_GE(num_devices, 1, TF_Message(status));
            for (int i = 0; i < num_devices; ++i)
            {
                EXPECT_NE("", c_api.TF_DeviceListName(devices, i, status), TF_Message(status));
                EXPECT_EQ(TF_OK, TF_GetCode(status), TF_Message(status));
            }

            c_api.TF_DeleteDeviceList(devices);
        }
    }
}