sglang_v0.5.2/pytorch_2.8.0/third_party/XNNPACK/emscripten.bzl

44 lines
1.3 KiB
Python

"""Emscripten-specific build definitions for XNNPACK."""
def xnnpack_emscripten_minimal_linkopts():
"""Minimal Emscripten-specific linkopts for binaries."""
return [
"-s ASSERTIONS=0",
"-s ENVIRONMENT=node,shell,web",
"-s ERROR_ON_UNDEFINED_SYMBOLS=1",
"-s EXIT_RUNTIME=1",
]
def xnnpack_emscripten_test_linkopts():
"""Emscripten-specific linkopts for unit tests."""
return [
"-s ALLOW_MEMORY_GROWTH=1",
"-s ASSERTIONS=2",
"-s ENVIRONMENT=node,shell,web",
"-s ERROR_ON_UNDEFINED_SYMBOLS=1",
"-s EXIT_RUNTIME=1",
"-s STACK_SIZE=5MB",
"-s TEXTDECODER=1",
"--pre-js $(location //:preamble.js.lds)",
]
def xnnpack_emscripten_benchmark_linkopts():
"""Emscripten-specific linkopts for benchmarks."""
return [
"-s ASSERTIONS=1",
"-s ENVIRONMENT=node,shell,web",
"-s ERROR_ON_UNDEFINED_SYMBOLS=1",
"-s EXIT_RUNTIME=1",
"-s ALLOW_MEMORY_GROWTH=1",
"-s INITIAL_MEMORY=1gb",
"-s MAXIMUM_MEMORY=4gb",
"-s TEXTDECODER=1",
"--pre-js $(location //:preamble.js.lds)",
]
def xnnpack_emscripten_deps():
"""Emscripten-specific dependencies for unit tests and benchmarks."""
return [
"//:preamble.js.lds",
]