FilesExpand file tree

 main

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

24 lines (21 loc) · 748 Bytes

 main

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

24 lines (21 loc) · 748 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

# Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.

#

# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE

import ast

from setuptools import setup

# We want to keep the version in sync with cuda.bindings, but setuptools would not let

# us to refer to any files outside of the project root, so we have to employ our own

# run-time lookup using setup()...

with open("../cuda_bindings/cuda/bindings/_version.py") as f:

for line in f:

if line.startswith("__version__"):

version = ast.parse(line).body[0].value.value

setup(

version=version,

install_requires=[

f"cuda-bindings~={version}",

],

extras_require={

"all": [f"cuda-bindings[all]~={version}"],

},

)