diff --git a/include/triton/common/triton_json.h b/include/triton/common/triton_json.h index 2b439e9..73900a5 100644 --- a/include/triton/common/triton_json.h +++ b/include/triton/common/triton_json.h @@ -1,4 +1,4 @@ -// Copyright 2020-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +// Copyright 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions @@ -47,6 +47,7 @@ #include #include #include +#include #include #include @@ -177,7 +178,8 @@ class TritonJsonImpl { TRITONJSON_STATUSRETURN( std::string("JSON parsing only available for top-level document")); } - const unsigned int parseFlags = rapidjson::kParseNanAndInfFlag; + const unsigned int parseFlags = + rapidjson::kParseNanAndInfFlag | rapidjson::kParseIterativeFlag; document_.Parse(base, size); if (document_.HasParseError()) { TRITONJSON_STATUSRETURN(std::string(