davrot/pytorch-sbs
1
0
Fork 0
mirror of https://github.com/davrot/pytorch-sbs.git synced 2025-07-06 11:00:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-02-02 19:15:23 +01:00 • committed by GitHub
parent 3622b91591
commit ec40834df1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
31 changed files with 2707 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

40
network/multiplication_approximation_gpu_cpp/Makefile Normal file
View file

@ -0,0 +1,40 @@
include ../.env
export
name = MultiplicationApproximation
type = GPU
PYPOSTFIX := $(shell $(PYBIN)python3-config --extension-suffix)
PYBIND11INCLUDE := $(shell $(PYBIN)python3 -m pybind11 --includes)
PARAMETERS_O = $(PARAMETERS_O_GPU) $(PYBIND11INCLUDE)
PARAMETERS_Linker = $(PARAMETERS_Linker_GPU)
so_file = Py$(name)$(type)$(PYPOSTFIX)
pyi_file = Py$(name)$(type).pyi
all: ../$(so_file)
$(O_DIRS)$(name)$(type).o: $(name)$(type).h \
$(name)$(type).cu \
gpu_error_term.cu \
gpu_approximation_multiplication_function.cu
mkdir -p $(O_DIRS)
$(NVCC) $(PARAMETERS_O) -c $(name)$(type).cu -o $(O_DIRS)$(name)$(type).o
$(O_DIRS)Py$(name)$(type).o: $(name)$(type).h Py$(name)$(type).cpp
mkdir -p $(O_DIRS)
$(NVCC) $(PARAMETERS_O) -c Py$(name)$(type).cpp -o $(O_DIRS)Py$(name)$(type).o
../$(so_file): \
$(O_DIRS)$(name)$(type).o \
$(O_DIRS)Py$(name)$(type).o
$(NVCC) $(PARAMETERS_Linker) -o ../$(so_file) \
$(O_DIRS)$(name)$(type).o \
$(O_DIRS)Py$(name)$(type).o
#######################
clean:
rm -rf $(O_DIRS)
rm -f ../$(so_file)
rm -f ../$(pyi_file)