|
1 | 1 | /*
|
2 |
| - * Copyright (c) 2023-2024, NVIDIA CORPORATION. |
| 2 | + * Copyright (c) 2023-2025, NVIDIA CORPORATION. |
3 | 3 | *
|
4 | 4 | * Licensed under the Apache License, Version 2.0 (the "License");
|
5 | 5 | * you may not use this file except in compliance with the License.
|
@@ -381,39 +381,55 @@ using preset_decision_forest = decision_forest<
|
381 | 381 | } // namespace detail
|
382 | 382 |
|
383 | 383 | /** A variant containing all standard decision_forest instantiations */
|
384 |
| -using decision_forest_variant = |
385 |
| - std::variant<detail::preset_decision_forest< |
386 |
| - std::variant_alternative_t<0, detail::specialization_variant>::layout, |
387 |
| - std::variant_alternative_t<0, detail::specialization_variant>::is_double_precision, |
388 |
| - std::variant_alternative_t<0, detail::specialization_variant>::has_large_trees>, |
389 |
| - detail::preset_decision_forest< |
390 |
| - std::variant_alternative_t<1, detail::specialization_variant>::layout, |
391 |
| - std::variant_alternative_t<1, detail::specialization_variant>::is_double_precision, |
392 |
| - std::variant_alternative_t<1, detail::specialization_variant>::has_large_trees>, |
393 |
| - detail::preset_decision_forest< |
394 |
| - std::variant_alternative_t<2, detail::specialization_variant>::layout, |
395 |
| - std::variant_alternative_t<2, detail::specialization_variant>::is_double_precision, |
396 |
| - std::variant_alternative_t<2, detail::specialization_variant>::has_large_trees>, |
397 |
| - detail::preset_decision_forest< |
398 |
| - std::variant_alternative_t<3, detail::specialization_variant>::layout, |
399 |
| - std::variant_alternative_t<3, detail::specialization_variant>::is_double_precision, |
400 |
| - std::variant_alternative_t<3, detail::specialization_variant>::has_large_trees>, |
401 |
| - detail::preset_decision_forest< |
402 |
| - std::variant_alternative_t<4, detail::specialization_variant>::layout, |
403 |
| - std::variant_alternative_t<4, detail::specialization_variant>::is_double_precision, |
404 |
| - std::variant_alternative_t<4, detail::specialization_variant>::has_large_trees>, |
405 |
| - detail::preset_decision_forest< |
406 |
| - std::variant_alternative_t<5, detail::specialization_variant>::layout, |
407 |
| - std::variant_alternative_t<5, detail::specialization_variant>::is_double_precision, |
408 |
| - std::variant_alternative_t<5, detail::specialization_variant>::has_large_trees>, |
409 |
| - detail::preset_decision_forest< |
410 |
| - std::variant_alternative_t<6, detail::specialization_variant>::layout, |
411 |
| - std::variant_alternative_t<6, detail::specialization_variant>::is_double_precision, |
412 |
| - std::variant_alternative_t<6, detail::specialization_variant>::has_large_trees>, |
413 |
| - detail::preset_decision_forest< |
414 |
| - std::variant_alternative_t<7, detail::specialization_variant>::layout, |
415 |
| - std::variant_alternative_t<7, detail::specialization_variant>::is_double_precision, |
416 |
| - std::variant_alternative_t<7, detail::specialization_variant>::has_large_trees>>; |
| 384 | +using decision_forest_variant = std::variant< |
| 385 | + detail::preset_decision_forest< |
| 386 | + std::variant_alternative_t<0, detail::specialization_variant>::layout, |
| 387 | + std::variant_alternative_t<0, detail::specialization_variant>::is_double_precision, |
| 388 | + std::variant_alternative_t<0, detail::specialization_variant>::has_large_trees>, |
| 389 | + detail::preset_decision_forest< |
| 390 | + std::variant_alternative_t<1, detail::specialization_variant>::layout, |
| 391 | + std::variant_alternative_t<1, detail::specialization_variant>::is_double_precision, |
| 392 | + std::variant_alternative_t<1, detail::specialization_variant>::has_large_trees>, |
| 393 | + detail::preset_decision_forest< |
| 394 | + std::variant_alternative_t<2, detail::specialization_variant>::layout, |
| 395 | + std::variant_alternative_t<2, detail::specialization_variant>::is_double_precision, |
| 396 | + std::variant_alternative_t<2, detail::specialization_variant>::has_large_trees>, |
| 397 | + detail::preset_decision_forest< |
| 398 | + std::variant_alternative_t<3, detail::specialization_variant>::layout, |
| 399 | + std::variant_alternative_t<3, detail::specialization_variant>::is_double_precision, |
| 400 | + std::variant_alternative_t<3, detail::specialization_variant>::has_large_trees>, |
| 401 | + detail::preset_decision_forest< |
| 402 | + std::variant_alternative_t<4, detail::specialization_variant>::layout, |
| 403 | + std::variant_alternative_t<4, detail::specialization_variant>::is_double_precision, |
| 404 | + std::variant_alternative_t<4, detail::specialization_variant>::has_large_trees>, |
| 405 | + detail::preset_decision_forest< |
| 406 | + std::variant_alternative_t<5, detail::specialization_variant>::layout, |
| 407 | + std::variant_alternative_t<5, detail::specialization_variant>::is_double_precision, |
| 408 | + std::variant_alternative_t<5, detail::specialization_variant>::has_large_trees>, |
| 409 | + detail::preset_decision_forest< |
| 410 | + std::variant_alternative_t<6, detail::specialization_variant>::layout, |
| 411 | + std::variant_alternative_t<6, detail::specialization_variant>::is_double_precision, |
| 412 | + std::variant_alternative_t<6, detail::specialization_variant>::has_large_trees>, |
| 413 | + detail::preset_decision_forest< |
| 414 | + std::variant_alternative_t<7, detail::specialization_variant>::layout, |
| 415 | + std::variant_alternative_t<7, detail::specialization_variant>::is_double_precision, |
| 416 | + std::variant_alternative_t<7, detail::specialization_variant>::has_large_trees>, |
| 417 | + detail::preset_decision_forest< |
| 418 | + std::variant_alternative_t<8, detail::specialization_variant>::layout, |
| 419 | + std::variant_alternative_t<8, detail::specialization_variant>::is_double_precision, |
| 420 | + std::variant_alternative_t<8, detail::specialization_variant>::has_large_trees>, |
| 421 | + detail::preset_decision_forest< |
| 422 | + std::variant_alternative_t<9, detail::specialization_variant>::layout, |
| 423 | + std::variant_alternative_t<9, detail::specialization_variant>::is_double_precision, |
| 424 | + std::variant_alternative_t<9, detail::specialization_variant>::has_large_trees>, |
| 425 | + detail::preset_decision_forest< |
| 426 | + std::variant_alternative_t<10, detail::specialization_variant>::layout, |
| 427 | + std::variant_alternative_t<10, detail::specialization_variant>::is_double_precision, |
| 428 | + std::variant_alternative_t<10, detail::specialization_variant>::has_large_trees>, |
| 429 | + detail::preset_decision_forest< |
| 430 | + std::variant_alternative_t<11, detail::specialization_variant>::layout, |
| 431 | + std::variant_alternative_t<11, detail::specialization_variant>::is_double_precision, |
| 432 | + std::variant_alternative_t<11, detail::specialization_variant>::has_large_trees>>; |
417 | 433 |
|
418 | 434 | /**
|
419 | 435 | * Determine the variant index of the decision_forest type to used based on
|
|
0 commit comments