cahlen commited on
Commit
c191cdf
·
verified ·
1 Parent(s): 7363b01

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 _erdos_straus_cuda import *
 
13
  return mod
14
 
15
  _ext = _load_extension()
 
 
16
  from _erdos_straus_cuda import *
build/torch211-cxx11-cu130-x86_64-linux/_erdos_straus_cuda.so CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:51afa51c7efd60e82f3d8bf20d869beff8d1f6688317abf9e47c4eb6283283e4
3
- size 310824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec57a8e643f105814a8328b5c0b05a00adc2d22bf419c23f6ea23f86c60cda50
3
+ size 306728
build/torch211-cxx11-cu130-x86_64-linux/erdos_straus/__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