Function TEST_F

tests/test_tensor_debug.cpp:133–173

// ============================================================================

TEST_F(TensorDebugTest, Sort_Float32_Ascending) {
    std::cout << "\n=== TEST: Sort Float32 Ascending ===" << std::endl;

    auto torch_data = torch::tensor({3.0f, 1.0f, 4.0f, 1.0f, 5.0f}, torch::kCUDA);
    auto gs_data = torch_to_tensor(torch_data);

    std::cout << "Input: [3.0, 1.0, 4.0, 1.0, 5.0]" << std::endl;

    // LibTorch
    auto [torch_vals, torch_idx] = torch::sort(torch_data);
    std::cout << "LibTorch result:" << std::endl;
    std::cout << "  Values: " << torch_vals << std::endl;
    std::cout << "  Indices: " << torch_idx << std::endl;
    std::cout << "  Index dtype: " << torch_idx.dtype() << std::endl;

    // lfs::core::Tensor
    std::cout << "lfs::core::Tensor result:" << std::endl;
    try {
        auto result = gs_data.sort(0);
        auto gs_vals = result.first;
        auto gs_idx = result.second;

        std::cout << "  Values shape: [" << gs_vals.shape()[0] << "]" << std::endl;
        std::cout << "  Indices shape: [" << gs_idx.shape()[0] << "]" << std::endl;
        std::cout << "  Index dtype: " << lfs::core::dtype_name(gs_idx.dtype()) << std::endl;

        // Compare values
        bool vals_match = compare_float_tensors(gs_vals, torch_vals);
        print_comparison("Sort values match", vals_match);
        EXPECT_TRUE(vals_match);

        // Compare indices (allow Int32 or Int64)
        bool idx_match = compare_int_tensors(gs_idx, torch_idx);
        print_comparison("Sort indices match", idx_match);
        EXPECT_TRUE(idx_match);

    } catch (const std::exception& e) {
        std::cout << "  ERROR: " << e.what() << std::endl;
        FAIL() << "lfs::core::Tensor sort() threw exception";
    }
}

TEST_F(TensorDebugTest, Sort_Float32_Descending) {
    std::cout << "\n=== TEST: Sort Float32 Descending ===" << std::endl;