code: def get_wheel_url():
      # Determine the version numbers that will be used to determine the correct wheel
      # We're using the CUDA version used to build torch, not the one currently installed
      # _, cuda_version_raw = get_cuda_bare_metal_version(CUDA_HOME)
      torch_cuda_version = parse(torch.version.cuda)
      torch_version_raw = parse(torch.__version__)
      # For CUDA 11, we only compile for CUDA 11.8, and for CUDA 12 we only compile for CUDA 12.2
      # to save CI time. Minor versions should be compatible.
      torch_cuda_version = (
          parse("11.8")
          if torch_cuda_version.major == 11
          else parse("12.2")
      )
      python_version = (
          f"cp{sys.version_info.major}{sys.version_info.minor}"
      )
      platform_name = get_platform()
      flash_version = get_package_version()
      # cuda_version = f"{cuda_version_raw.major}{cuda_version_raw.minor}"
      cuda_version = (
          f"{torch_cuda_version.major}{torch_cuda_version.minor}"
      )
      torch_version = (
          f"{torch_version_raw.major}.{torch_version_raw.minor}"
      )
      cxx11_abi = str(torch._C._GLIBCXX_USE_CXX11_ABI).upper()
  
      # Determine wheel URL based on CUDA version, torch version, python version and OS
      wheel_filename = f"{PACKAGE_NAME}-{flash_version}+cu{cuda_version}torch{torch_version}cxx11abi{cxx11_abi}-{python_version}-{python_version}-{platform_name}.whl"
      wheel_url = BASE_WHEEL_URL.format(
          tag_name=f"v{flash_version}", wheel_name=wheel_filename
      )
      return wheel_url, wheel_filename
  
imports: import sys
  import warnings
  import os
  import re
  import ast
  from pathlib import Path
  from packaging.version import parse, Version
  import platform
  from setuptools import setup, find_packages
  import subprocess
  import urllib.request
  import urllib.error
  from wheel.bdist_wheel import bdist_wheel
  import torch
  from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension, CUDA_HOME
label_desc: Use image node_id 6 for calling a global function with example usage: get_wheel_url() and returns: wheel_url,  wheel_filename