CC = nvcc
CFLAGS = 
fPIC_flag = -Xcompiler=-fPIC

obj_files = conv2d_bias_1541.obj conv2d_bias_1542.obj groupnorm_swish_1543.obj conv2d_bias_1544.obj conv2d_bias_add_identity_1638.obj groupnorm_1548.obj gemm_rcr_bias_permute_1551.obj split_1553.obj reshape_1554.obj mem_eff_attention_1557.obj gemm_rcr_bias_add_1559.obj fused_elementwise_1652.obj upsampling2d_1583.obj conv2d_bias_1584.obj groupnorm_swish_1585.obj conv2d_bias_add_identity_1643.obj upsampling2d_1600.obj conv2d_bias_1601.obj groupnorm_swish_1602.obj conv2d_bias_1603.obj groupnorm_swish_1604.obj conv2d_bias_1606.obj conv2d_bias_add_identity_1646.obj conv2d_bias_1609.obj upsampling2d_1618.obj conv2d_bias_1619.obj groupnorm_swish_1620.obj conv2d_bias_1621.obj groupnorm_swish_1622.obj conv2d_bias_1624.obj conv2d_bias_add_identity_1649.obj conv2d_bias_1627.obj conv2d_bias_1637.obj constants.obj model_container_base.obj debug_utility.obj model_container.obj model_interface.obj utility.obj

%.obj : %.cu
	nvcc -DCUTLASS_ENABLE_TENSOR_CORE_MMA=1 -DCUTLASS_USE_TANH_FOR_SIGMOID=1 -w -gencode=arch=compute_80,code=[sm_80,compute_80] -Xcompiler=-fPIC -Xcompiler=-Wconversion -Xcompiler=-fno-strict-aliasing -Xcompiler -fvisibility=hidden -O3 -std=c++17 --expt-relaxed-constexpr --use_fast_math -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/include -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/tools/util/include -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/examples/35_gemm_softmax -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/examples/41_fused_multi_head_attention -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/examples/45_dual_gemm -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/../../backend/cuda/attention/src/./ -I/home/olab/kirstain/anaconda3/envs/ait/lib/python3.10/site-packages/aitemplate/3rdparty/cutlass/../../backend/cuda/attention/src/fmha -DNDEBUG -c -o $@ $<
%.obj : %.bin
	ld -r -b binary -o $@ $<

.PHONY: all clean clean_constants
all: test.so

test.so: $(obj_files)
	$(CC) -shared $(fPIC_flag) $(CFLAGS) -o $@ $(obj_files)



clean:
	rm -f *.obj test.so

clean_constants:
	rm -f constants.bin