cahlen commited on
Commit
4fef8c4
·
verified ·
1 Parent(s): 2807b6c

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 _flint_hills_cuda import *
 
13
  return mod
14
 
15
  _ext = _load_extension()
 
 
16
  from _flint_hills_cuda import *
build/torch211-cxx11-cu130-x86_64-linux/_flint_hills_cuda.so CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7d44a06dd1689a8a7489eae018f150bbace26194fc793ec89e7ac337aeba3c98
3
- size 648992
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1d99ec1c45df080c6ea7c97e2493dd2d97645915f78a5950be86295afbb88a8
3
+ size 763240
build/torch211-cxx11-cu130-x86_64-linux/flint_hills/__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