mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
6dae1e419e
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76376 This is only used in a few places and only aliases the c10 macros header. Differential Revision: [D35904936](https://our.internmc.facebook.com/intern/diff/D35904936/) Approved by: https://github.com/dreiss, https://github.com/malfet
53 lines
1.4 KiB
C++
53 lines
1.4 KiB
C++
// Taken from
|
|
// https://github.com/JuliaLang/julia/blob/v1.1.0/src/support/strtod.c
|
|
#include <torch/csrc/jit/frontend/strtod.h>
|
|
|
|
#include <c10/macros/Macros.h>
|
|
#include <clocale>
|
|
#include <cstdlib>
|
|
|
|
#if defined(__APPLE__) || defined(__FreeBSD__)
|
|
#include <xlocale.h>
|
|
#endif
|
|
|
|
// The following code is derived from the Python function _PyOS_ascii_strtod
|
|
// see http://hg.python.org/cpython/file/default/Python/pystrtod.c
|
|
//
|
|
// Copyright © 2001-2014 Python Software Foundation; All Rights Reserved
|
|
//
|
|
// The following modifications have been made:
|
|
// - Leading spaces are ignored
|
|
// - Parsing of hex floats is supported in the derived version
|
|
// - Python functions for tolower, isdigit and malloc have been replaced by the
|
|
// respective
|
|
// C stdlib functions
|
|
|
|
#include <cctype>
|
|
#include <cerrno>
|
|
#include <cmath>
|
|
#include <cstring>
|
|
#include <locale>
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
#ifdef _MSC_VER
|
|
double strtod_c(const char* nptr, char** endptr) {
|
|
static _locale_t loc = _create_locale(LC_ALL, "C");
|
|
return _strtod_l(nptr, endptr, loc);
|
|
}
|
|
#else
|
|
double strtod_c(const char* nptr, char** endptr) {
|
|
/// NOLINTNEXTLINE(hicpp-signed-bitwise)
|
|
static locale_t loc = newlocale(LC_ALL_MASK, "C", nullptr);
|
|
return strtod_l(nptr, endptr, loc);
|
|
}
|
|
#endif
|
|
|
|
float strtof_c(const char* nptr, char** endptr) {
|
|
return (float)strtod_c(nptr, endptr);
|
|
}
|
|
|
|
} // namespace jit
|
|
} // namespace torch
|