cahlen commited on
Commit
e867335
·
verified ·
1 Parent(s): 7527b77

Add torch bindings + build variant torch211-cxx11-cu130-x86_64-linux

Browse files
build/torch211-cxx11-cu130-x86_64-linux/__init__.py CHANGED
@@ -13,6 +13,4 @@ def _load_extension():
13
  return mod
14
 
15
  _ext = _load_extension()
16
-
17
- # Re-export all public symbols from the extension
18
  from _class_numbers_cuda import *
 
13
  return mod
14
 
15
  _ext = _load_extension()
 
 
16
  from _class_numbers_cuda import *
build/torch211-cxx11-cu130-x86_64-linux/_class_numbers_cuda.so CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fdd606109e55d749c6f15fdb6b89f4eda5aa06e78df4586b137e33e52e498f88
3
- size 645896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc23811eed8b5ebdd11641dc4331bdc28c6716ce82c5767152300ac85cff4879
3
+ size 694216
build/torch211-cxx11-cu130-x86_64-linux/class_numbers/__init__.py CHANGED
@@ -3,12 +3,11 @@ from pathlib import Path
3
 
4
  def _import_from_path(file_path):
5
  path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
6
- module_name = path_hash
7
- spec = importlib.util.spec_from_file_location(module_name, file_path)
8
  if spec is None:
9
- raise ImportError(f"Cannot load spec for {module_name} from {file_path}")
10
  module = importlib.util.module_from_spec(spec)
11
- sys.modules[module_name] = module
12
  spec.loader.exec_module(module)
13
  return module
14
 
 
3
 
4
  def _import_from_path(file_path):
5
  path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
6
+ spec = importlib.util.spec_from_file_location(path_hash, file_path)
 
7
  if spec is None:
8
+ raise ImportError(f"Cannot load spec from {file_path}")
9
  module = importlib.util.module_from_spec(spec)
10
+ sys.modules[path_hash] = module
11
  spec.loader.exec_module(module)
12
  return module
13