cahlen commited on
Commit
bee95b7
·
verified ·
1 Parent(s): f873e77

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 _kronecker_cuda import *
 
13
  return mod
14
 
15
  _ext = _load_extension()
 
 
16
  from _kronecker_cuda import *
build/torch211-cxx11-cu130-x86_64-linux/_kronecker_cuda.so CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0c3baca2427be738d8c5d576d26fb3f72b92d149ac7cef2445f94fd3c6e11ee9
3
- size 251352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b2472efa1e34953c929e1514277030eb82ba832ce30ca1c69095f202589d302
3
+ size 374448
build/torch211-cxx11-cu130-x86_64-linux/kronecker/__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