rsloop 0.1.44__tar.gz → 0.1.46__tar.gz

{rsloop-0.1.44 → rsloop-0.1.46}/.github/workflows/wheels.yml

@@ -62,7 +62,8 @@ jobs:
       - name: Set up Rust
         uses: dtolnay/rust-toolchain@stable
         with:
-          toolchain: 1.95.0
+          toolchain: stable
+          components: llvm-tools-preview
 
       - name: Cache Cargo registry and target
         uses: Swatinem/rust-cache@v2
@@ -72,7 +73,7 @@ jobs:
         shell: bash
         env:
           RSLOOP_PYTHON_VERSIONS: ${{ matrix.python_versions }}
-        run: scripts/build-wheels.sh --out dist --target "${{ matrix.rust_target }}"
+        run: scripts/build-pgo-wheels.sh --out dist --target "${{ matrix.rust_target }}"
 
       - name: Upload wheel artifacts
         uses: actions/upload-artifact@v4
@@ -102,7 +103,7 @@ jobs:
       - name: Set up Rust
         uses: dtolnay/rust-toolchain@stable
         with:
-          toolchain: 1.95.0
+          toolchain: stable
 
       - name: Cache Cargo registry and target
         uses: Swatinem/rust-cache@v2

{rsloop-0.1.44 → rsloop-0.1.46}/Cargo.lock

@@ -197,15 +197,9 @@ dependencies = [
 
 [[package]]
 name = "bitflags"
-version = "1.3.2"
+version = "2.13.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
-
-[[package]]
-name = "bitflags"
-version = "2.13.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "b4388bee8683e3d04af747c73422af53102d2bd24d9eadb6cbc100baef4b43f8"
+checksum = "b588b76d00fde79687d7646a9b5bdf3cc0f655e0bbd080335a95d7e96f3587da"
 
 [[package]]
 name = "blocking"
@@ -522,11 +516,12 @@ checksum = "fc0fef456e4baa96da950455cd02c081ca953b141298e41db3fc7e36b1da849c"
 
 [[package]]
 name = "io-uring"
-version = "0.6.4"
+version = "0.7.14"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "595a0399f411a508feb2ec1e970a4a30c249351e30208960d58298de8660b0e5"
+checksum = "d64d8ca234d152948ceaede1f419b6a83983a5ecccaac05fb337a809c96d3aa6"
 dependencies = [
- "bitflags 1.3.2",
+ "bitflags",
+ "cfg-if",
  "libc",
 ]
 
@@ -568,9 +563,9 @@ checksum = "bbd2bcb4c963f2ddae06a2efc7e9f3591312473c50c6685e1f298068316e66fe"
 
 [[package]]
 name = "libc"
-version = "0.2.186"
+version = "0.2.189"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "68ab91017fe16c622486840e4c83c9a37afeff978bd239b5293d61ece587de66"
+checksum = "3eaf3ede3fee6db1a4c2ee091bf8a8b4dccdc6d17f656fb07896ee72867612f2"
 
 [[package]]
 name = "linux-raw-sys"
@@ -730,33 +725,13 @@ dependencies = [
  "unicode-ident",
 ]
 
-[[package]]
-name = "profiling"
-version = "1.0.18"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "3d595e54a326bc53c1c197b32d295e14b169e3cfeaa8dc82b529f947fba6bcf5"
-dependencies = [
- "profiling-procmacros",
- "tracy-client",
-]
-
-[[package]]
-name = "profiling-procmacros"
-version = "1.0.18"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "4488a4a36b9a4ba6b9334a32a39971f77c1436ec82c38707bce707699cc3bbcb"
-dependencies = [
- "quote",
- "syn",
-]
-
 [[package]]
 name = "proptest"
 version = "1.11.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "4b45fcc2344c680f5025fe57779faef368840d0bd1f42f216291f0dc4ace4744"
 dependencies = [
- "bitflags 2.13.0",
+ "bitflags",
  "num-traits",
  "rand",
  "rand_chacha",
@@ -922,15 +897,19 @@ dependencies = [
 
 [[package]]
 name = "rsloop"
-version = "0.1.44"
+version = "0.1.46"
 dependencies = [
  "async-io",
  "async-std",
  "crossbeam-channel",
+ "crossbeam-queue",
  "futures",
+ "futures-util",
+ "io-uring",
  "libc",
  "memchr",
- "profiling",
+ "mio",
+ "oneshot",
  "proptest",
  "pyo3",
  "pyo3-async-runtimes",
@@ -939,10 +918,10 @@ dependencies = [
  "rustls-native-certs",
  "rustls-pemfile",
  "signal-hook",
+ "slab",
  "socket2",
  "tokio",
  "tracy-client",
- "vibeio",
  "windows-sys 0.61.2",
 ]
 
@@ -952,7 +931,7 @@ version = "1.1.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b6fe4565b9518b83ef4f91bb47ce29620ca828bd32cb7e408f0062e9930ba190"
 dependencies = [
- "bitflags 2.13.0",
+ "bitflags",
  "errno",
  "libc",
  "linux-raw-sys",
@@ -1043,7 +1022,7 @@ version = "3.7.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b7f4bc775c73d9a02cde8bf7b2ec4c9d12743edf609006c7facc23998404cd1d"
 dependencies = [
- "bitflags 2.13.0",
+ "bitflags",
  "core-foundation",
  "core-foundation-sys",
  "libc",
@@ -1259,21 +1238,6 @@ version = "1.12.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "7ba6f5989077681266825251a52748b8c1d8a4ad098cc37e440103d0ea717fc0"
 
-[[package]]
-name = "vibeio"
-version = "0.2.15"
-dependencies = [
- "crossbeam-queue",
- "futures-util",
- "io-uring",
- "libc",
- "mio",
- "oneshot",
- "slab",
- "tokio",
- "windows-sys 0.61.2",
-]
-
 [[package]]
 name = "wasi"
 version = "0.11.1+wasi-snapshot-preview1"

{rsloop-0.1.44 → rsloop-0.1.46}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "rsloop"
-version = "0.1.44"
+version = "0.1.46"
 edition = "2024"
 description = "An event loop for asyncio written in Rust"
 license = "Apache-2.0"
@@ -15,29 +15,37 @@ crate-type = ["cdylib", "rlib"]
 
 [features]
 default = []
-profiler = ["dep:tracy-client", "profiling/profile-with-tracy"]
+profiler = ["dep:tracy-client"]
 
 [dependencies]
 async-io = "2"
 async-std = "1"
 crossbeam-channel = "0.5"
+crossbeam-queue = "0.3"
 futures = { version = "0.3", default-features = false, features = ["std"] }
+futures-util = "0.3"
 libc = "0.2"
 memchr = "2"
-profiling = "1"
+mio = "1"
+oneshot = { version = "0.2", features = ["async", "std"] }
 pyo3 = { version = "0.29.2" }
 pyo3-async-runtimes = { version = "0.29", features = ["async-std-runtime"] }
 rustls = "0.23"
 rustls-native-certs = "0.8"
 rustls-pemfile = "2"
 signal-hook = "0.4"
+slab = "0.4"
 socket2 = "0.6"
 tokio = { version = "1", default-features = false, features = ["io-util"] }
 tracy-client = { version = "0.18", default-features = false, features = ["enable", "only-localhost", "sampling"], optional = true }
-vibeio = { version = "0.2.15", path = "vendor/vibeio", default-features = false, features = ["time"] }
+[target.'cfg(target_os = "linux")'.dependencies]
+io-uring = "0.7"
+
+[target.'cfg(unix)'.dependencies]
+mio = { version = "1", features = ["os-poll", "os-ext"] }
 
 [target.'cfg(windows)'.dependencies]
-windows-sys = { version = "0.61", features = ["Win32_Foundation", "Win32_Networking_WinSock", "Win32_System_IO", "Win32_System_Pipes", "Win32_System_Threading"] }
+windows-sys = { version = "0.61", features = ["Wdk_Foundation", "Wdk_Storage_FileSystem", "Wdk_System_IO", "Win32_Foundation", "Win32_Networking_WinSock", "Win32_Security", "Win32_Storage_FileSystem", "Win32_System_Console", "Win32_System_IO", "Win32_System_Pipes", "Win32_System_Threading"] }
 
 [build-dependencies]
 pyo3-build-config = "0.29.2"
@@ -83,7 +91,7 @@ borrow_as_ptr = "allow"
 ref_as_ptr = "allow"
 
 [lints.rust]
-unexpected_cfgs = { level = "warn", check-cfg = ["cfg(kani)"] }
+unexpected_cfgs = { level = "warn", check-cfg = ["cfg(kani)", "cfg(musl_v1_2_3)", "cfg(syscall_accept4)", "cfg(syscall_pipe2)", "cfg(feature, values(\"blocking-default\", \"fs\", \"pipe\", \"process\", \"signal\", \"splice\", \"stdio\"))"] }
 
 [package.metadata.kani.flags]
 # Force every proof that reaches a loop to state and review its own bound.

{rsloop-0.1.44 → rsloop-0.1.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rsloop
-Version: 0.1.44
+Version: 0.1.46
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
@@ -379,6 +379,27 @@ Build release wheels into `dist/wheels`:
 scripts/build-wheels.sh
 ```
 
+Build the published-wheel configuration with profile-guided optimization:
+
+```bash
+rustup component add llvm-tools-preview
+scripts/build-pgo-wheels.sh
+```
+
+For each requested Python ABI, the PGO wrapper creates an instrumented wheel,
+trains it on sustained HTTP, TLS, WebSocket, mixed-stream, bulk-transfer,
+idle-connection, callback, task, and TCP workloads, merges the resulting LLVM
+profiles, and builds that ABI's final wheel with its matching profile. Per-ABI
+training avoids discarding counters when PyO3's generated control flow differs
+between Python versions or free-threaded builds. The target must be native
+because the instrumented extension runs during training.
+
+Set `RSLOOP_PGO_SCENARIOS` to override the comma-separated network scenarios.
+Tagged and manually dispatched wheel workflows use PGO on every supported
+platform except Windows ARM64. Rust profile-generation binaries currently
+crash on that target ([rust-lang/rust#156675](https://github.com/rust-lang/rust/issues/156675)),
+so it temporarily falls back to the normal fat-LTO release build.
+
 [`scripts/build-wheels.sh`](./scripts/build-wheels.sh) currently defaults to
 CPython `3.10 3.11 3.12 3.13 3.14`, and
 uses `uv python install` / `uv python find` to locate interpreters.

{rsloop-0.1.44 → rsloop-0.1.46}/README.md

@@ -347,6 +347,27 @@ Build release wheels into `dist/wheels`:
 scripts/build-wheels.sh
 ```
 
+Build the published-wheel configuration with profile-guided optimization:
+
+```bash
+rustup component add llvm-tools-preview
+scripts/build-pgo-wheels.sh
+```
+
+For each requested Python ABI, the PGO wrapper creates an instrumented wheel,
+trains it on sustained HTTP, TLS, WebSocket, mixed-stream, bulk-transfer,
+idle-connection, callback, task, and TCP workloads, merges the resulting LLVM
+profiles, and builds that ABI's final wheel with its matching profile. Per-ABI
+training avoids discarding counters when PyO3's generated control flow differs
+between Python versions or free-threaded builds. The target must be native
+because the instrumented extension runs during training.
+
+Set `RSLOOP_PGO_SCENARIOS` to override the comma-separated network scenarios.
+Tagged and manually dispatched wheel workflows use PGO on every supported
+platform except Windows ARM64. Rust profile-generation binaries currently
+crash on that target ([rust-lang/rust#156675](https://github.com/rust-lang/rust/issues/156675)),
+so it temporarily falls back to the normal fat-LTO release build.
+
 [`scripts/build-wheels.sh`](./scripts/build-wheels.sh) currently defaults to
 CPython `3.10 3.11 3.12 3.13 3.14`, and
 uses `uv python install` / `uv python find` to locate interpreters.

{rsloop-0.1.44/vendor/vibeio → rsloop-0.1.46}/build.rs

@@ -1,4 +1,7 @@
 fn main() {
+    pyo3_build_config::use_pyo3_cfgs();
+    pyo3_build_config::add_extension_module_link_args();
+
     let musl = std::env::var("CARGO_CFG_TARGET_ENV").unwrap_or_default() == "musl";
     let musl_v1_2_3 = std::env::var("RUST_LIBC_UNSTABLE_MUSL_V1_2_3").is_ok();
     println!("cargo:rerun-if-env-changed=RUST_LIBC_UNSTABLE_MUSL_V1_2_3");

{rsloop-0.1.44 → rsloop-0.1.46}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "rsloop"
-version = "0.1.44"
+version = "0.1.46"
 description = "An event loop for asyncio written in Rust"
 readme = "README.md"
 license = { file = "LICENSE" }

rsloop-0.1.46/scripts/build-pgo-wheels.sh

@@ -0,0 +1,200 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
+source "${ROOT_DIR}/scripts/python-versions.sh"
+
+DEFAULT_PGO_SCENARIOS="http_keepalive,tls_http,websocket_messages,websocket_tls,mixed_streams,bulk_transfer,idle_connections"
+BUILD_WHEEL_ARGS=("$@")
+RUST_TARGET="${RSLOOP_RUST_TARGET:-}"
+INSTALL_PYTHONS=1
+MATURIN_ARGS=()
+
+usage() {
+  cat <<'EOF'
+Build PGO-optimized release wheels for rsloop.
+
+Usage:
+  scripts/build-pgo-wheels.sh [build-wheels options] [-- maturin args...]
+
+This wrapper accepts the same options as scripts/build-wheels.sh. For each
+requested Python ABI, it builds an instrumented native wheel, trains it with
+representative traffic, merges the LLVM profiles, and builds that ABI's final
+wheel with its matching profile.
+
+Environment:
+  RSLOOP_PGO_SCENARIOS    Comma-separated workload-matrix training scenarios
+
+PGO requires a native target and the Rust llvm-tools-preview component.
+EOF
+}
+
+host_rust_target() {
+  rustc -vV | sed -n 's/^host: //p'
+}
+
+native_path() {
+  case "${OSTYPE:-}" in
+    msys*|cygwin*) cygpath -m "$1" ;;
+    *) printf '%s\n' "$1" ;;
+  esac
+}
+
+shell_path() {
+  case "${OSTYPE:-}" in
+    msys*|cygwin*) cygpath -u "$1" ;;
+    *) printf '%s\n' "$1" ;;
+  esac
+}
+
+while (($#)); do
+  case "$1" in
+    -o|--out)
+      (($# >= 2)) || { echo "missing value for $1" >&2; exit 1; }
+      shift 2
+      ;;
+    -t|--target)
+      (($# >= 2)) || { echo "missing value for $1" >&2; exit 1; }
+      RUST_TARGET="$2"
+      shift 2
+      ;;
+    --skip-python-install)
+      INSTALL_PYTHONS=0
+      shift
+      ;;
+    -h|--help)
+      usage
+      printf '\nUnderlying wheel-builder options:\n\n'
+      "${ROOT_DIR}/scripts/build-wheels.sh" --help
+      exit 0
+      ;;
+    --)
+      shift
+      MATURIN_ARGS=("$@")
+      break
+      ;;
+    *)
+      echo "unknown argument: $1" >&2
+      usage >&2
+      exit 1
+      ;;
+  esac
+done
+
+cd "$ROOT_DIR"
+HOST_RUST_TARGET="$(host_rust_target)"
+if [[ -n "$RUST_TARGET" && "$RUST_TARGET" != "$HOST_RUST_TARGET" ]]; then
+  echo "PGO requires a native target because the instrumented wheel must run during training" >&2
+  echo "host target: ${HOST_RUST_TARGET}; requested target: ${RUST_TARGET}" >&2
+  exit 1
+fi
+if [[ "$RUST_TARGET" == *-apple-darwin ]]; then
+  export MACOSX_DEPLOYMENT_TARGET="${MACOSX_DEPLOYMENT_TARGET:-13.0}"
+fi
+if [[ "$RUST_TARGET" == "aarch64-pc-windows-msvc" ]]; then
+  # rustc profile-generate binaries currently crash on native Windows ARM64:
+  # https://github.com/rust-lang/rust/issues/156675
+  echo "PGO is unavailable on Windows ARM64; building a normal fat-LTO release wheel"
+  export RUSTFLAGS="${RUSTFLAGS:-} -Clink-arg=/arm64hazardfree"
+  exec "${ROOT_DIR}/scripts/build-wheels.sh" "${BUILD_WHEEL_ARGS[@]}"
+fi
+
+RSLOOP_PYTHON_VERSIONS_OVERRIDE="${RSLOOP_PYTHON_VERSIONS:-}"
+rsloop_load_python_versions "$RUST_TARGET"
+PYTHON_VERSIONS=("${RSLOOP_PYTHON_VERSIONS[@]}")
+PGO_SCENARIOS="${RSLOOP_PGO_SCENARIOS:-$DEFAULT_PGO_SCENARIOS}"
+
+rust_sysroot="$(shell_path "$(rustc --print sysroot)")"
+llvm_profdata="${rust_sysroot}/lib/rustlib/${HOST_RUST_TARGET}/bin/llvm-profdata"
+if [[ "${OSTYPE:-}" == msys* || "${OSTYPE:-}" == cygwin* ]]; then
+  llvm_profdata+=".exe"
+fi
+if [[ ! -x "$llvm_profdata" ]]; then
+  echo "llvm-profdata was not found at ${llvm_profdata}" >&2
+  echo "install it with: rustup component add llvm-tools-preview" >&2
+  exit 1
+fi
+
+PGO_WORK_DIR="$(mktemp -d "${TMPDIR:-/tmp}/rsloop-wheel-pgo.XXXXXX")"
+cleanup() {
+  if [[ -n "${PGO_WORK_DIR:-}" && -d "$PGO_WORK_DIR" ]]; then
+    rm -rf -- "$PGO_WORK_DIR"
+  fi
+}
+trap cleanup EXIT
+
+PROFILE_DIR="${PGO_WORK_DIR}/profiles"
+TARGET_GENERATE="$(native_path "${PGO_WORK_DIR}/target-generate")"
+TARGET_USE="$(native_path "${PGO_WORK_DIR}/target-use")"
+MERGED_PROFILE="${PGO_WORK_DIR}/merged.profdata"
+mkdir -p "$PROFILE_DIR"
+BASE_RUSTFLAGS="${RUSTFLAGS:-}"
+GENERATE_RUSTFLAGS="${BASE_RUSTFLAGS} -Cprofile-generate=$(native_path "$PROFILE_DIR")"
+USE_RUSTFLAGS="${BASE_RUSTFLAGS} -Cprofile-use=$(native_path "$MERGED_PROFILE") -Cllvm-args=-pgo-warn-missing-function"
+
+for version in "${PYTHON_VERSIONS[@]}"; do
+  python_request="$version"
+  if [[ -n "$RUST_TARGET" ]]; then
+    python_request="$(rsloop_target_python_request "$version" "$RUST_TARGET")"
+  fi
+  if (( INSTALL_PYTHONS )); then
+    uv python install "$python_request"
+  fi
+
+  version_key="${version//[^a-zA-Z0-9]/_}"
+  version_dir="${PGO_WORK_DIR}/${version_key}"
+  wheel_dir="${version_dir}/instrumented-wheel"
+  tls_dir="${version_dir}/tls"
+  training_venv="${version_dir}/venv"
+  mkdir -p "$wheel_dir"
+  find "$PROFILE_DIR" -maxdepth 1 -type f -name '*.profraw' -delete
+
+  instrumented_build=(uv run --no-project --python "$python_request" --with maturin maturin build --release --interpreter "$(uv python find "$python_request")" --out "$wheel_dir")
+  if [[ -n "$RUST_TARGET" ]]; then
+    instrumented_build+=(--target "$RUST_TARGET")
+  fi
+  if ((${#MATURIN_ARGS[@]})); then
+    instrumented_build+=("${MATURIN_ARGS[@]}")
+  fi
+  echo "Building instrumented PGO training wheel for Python ${version}"
+  CARGO_TARGET_DIR="$TARGET_GENERATE" RUSTFLAGS="$GENERATE_RUSTFLAGS" "${instrumented_build[@]}"
+
+  shopt -s nullglob
+  instrumented_wheels=("$wheel_dir"/*.whl)
+  shopt -u nullglob
+  if ((${#instrumented_wheels[@]} != 1)); then
+    echo "expected one instrumented training wheel, found ${#instrumented_wheels[@]}" >&2
+    exit 1
+  fi
+
+  uv venv --no-project --python "$python_request" "$training_venv"
+  if [[ -x "$training_venv/Scripts/python.exe" ]]; then
+    training_python="$training_venv/Scripts/python.exe"
+  else
+    training_python="$training_venv/bin/python"
+  fi
+  uv pip install --python "$training_python" "${instrumented_wheels[0]}"
+  "$training_python" scripts/generate_test_tls_certs.py "$tls_dir"
+
+  echo "Training Python ${version} PGO with sustained representative network traffic"
+  "$training_python" benches/workload_matrix.py --loops rsloop --scenarios "$PGO_SCENARIOS" --sustained --tls-dir "$tls_dir"
+  echo "Training Python ${version} PGO with callbacks, tasks, and TCP streams"
+  "$training_python" benches/compare_event_loops.py --loops rsloop
+
+  shopt -s nullglob
+  raw_profiles=("$PROFILE_DIR"/*.profraw)
+  shopt -u nullglob
+  if ((${#raw_profiles[@]} == 0)); then
+    echo "PGO training produced no .profraw files for Python ${version}" >&2
+    exit 1
+  fi
+
+  echo "Merging ${#raw_profiles[@]} LLVM profile shards for Python ${version}"
+  "$llvm_profdata" merge -o "$MERGED_PROFILE" "${raw_profiles[@]}"
+  echo "Building final Python ${version} wheel with its matching PGO profile"
+  RSLOOP_PYTHON_VERSIONS="$version" \
+    CARGO_TARGET_DIR="$TARGET_USE" \
+    RUSTFLAGS="$USE_RUSTFLAGS" \
+    "${ROOT_DIR}/scripts/build-wheels.sh" "${BUILD_WHEEL_ARGS[@]}"
+  rm -rf -- "$version_dir"
+done

{rsloop-0.1.44 → rsloop-0.1.46}/src/bindings/loop_api/connections.rs

@@ -37,7 +37,7 @@ pub(super) fn create_connection<'py>(
     py: Python<'py>,
     params: CreateConnectionParams,
 ) -> PyResult<Bound<'py, PyAny>> {
-    profiling::scope!("PyLoop::create_connection");
+    crate::profile_scope!("PyLoop::create_connection");
     let CreateConnectionParams {
         protocol_factory,
         host,
@@ -146,7 +146,7 @@ pub(super) fn create_connection_transport(
     sock: Py<PyAny>,
     tls: TlsParams,
 ) -> PyResult<Py<PyAny>> {
-    profiling::scope!("PyLoop::_create_connection_transport");
+    crate::profile_scope!("PyLoop::_create_connection_transport");
     tls.validate()?;
 
     let env = LoopSpawnEnv::capture(py, &slf)?;
@@ -214,7 +214,7 @@ pub(super) fn connect_accepted_socket<'py>(
     sock: Py<PyAny>,
     tls: TlsParams,
 ) -> PyResult<Bound<'py, PyAny>> {
-    profiling::scope!("PyLoop::connect_accepted_socket");
+    crate::profile_scope!("PyLoop::connect_accepted_socket");
     tls.validate()?;
 
     let locals = PyLoop::task_locals(py, &slf)?;
@@ -250,7 +250,7 @@ pub(super) fn start_tls<'py>(
     tls: TlsParams,
     server_side: bool,
 ) -> PyResult<Bound<'py, PyAny>> {
-    profiling::scope!("PyLoop::start_tls");
+    crate::profile_scope!("PyLoop::start_tls");
     let locals = PyLoop::task_locals(py, &slf)?;
     let locals_for_barrier = locals.clone();
     let transport: Py<PyStreamTransport> = transport.extract(py)?;

{rsloop-0.1.44 → rsloop-0.1.46}/src/bindings/loop_api/methods.rs

@@ -137,13 +137,13 @@ impl PyLoop {
         self.core.set_debug(enabled);
     }
 
-    #[profiling::function]
     fn run_forever(slf: Py<Self>, py: Python<'_>) -> PyResult<()> {
+        crate::profile_function!();
         lifecycle::run_forever(slf, py)
     }
 
-    #[profiling::function]
     fn run_until_complete(slf: Py<Self>, py: Python<'_>, future: Py<PyAny>) -> PyResult<Py<PyAny>> {
+        crate::profile_function!();
         lifecycle::run_until_complete(slf, py, future)
     }
 

{rsloop-0.1.44 → rsloop-0.1.46}/src/bindings/loop_api/servers.rs

@@ -42,7 +42,7 @@ pub(super) fn create_server<'py>(
     py: Python<'py>,
     params: CreateServerParams,
 ) -> PyResult<Bound<'py, PyAny>> {
-    profiling::scope!("PyLoop::create_server");
+    crate::profile_scope!("PyLoop::create_server");
     let CreateServerParams {
         protocol_factory,
         host,

{rsloop-0.1.44 → rsloop-0.1.46}/src/engine/callbacks.rs

@@ -129,7 +129,7 @@ impl ReadyCallback {
     /// A nested-context error falls back to direct invocation because that means
     /// the desired context is already active on this thread.
     pub fn invoke(&self, py: Python<'_>) -> PyResult<Py<PyAny>> {
-        profiling::scope!("ReadyCallback::invoke");
+        crate::profile_scope!("ReadyCallback::invoke");
         if !self.context_needs_run {
             return self.invoke_direct(py);
         }
@@ -152,7 +152,7 @@ impl ReadyCallback {
     }
 
     fn invoke_direct(&self, py: Python<'_>) -> PyResult<Py<PyAny>> {
-        profiling::scope!("ReadyCallback::invoke_direct");
+        crate::profile_scope!("ReadyCallback::invoke_direct");
         match &self.args {
             CallbackArgs::None => call_callback_noargs(py, &self.callback),
             CallbackArgs::One(arg) => call_callback_onearg(py, &self.callback, arg),