chore(sycl): add softmax forward, all kernels

mspronesti · Aug 1, 2024 · 9967fc0 · 9967fc0
1 parent 1bebd09
commit 9967fc0
Show file tree

Hide file tree

Showing 2 changed files with 564 additions and 1 deletion.
diff --git a/dev/sycl/Makefile b/dev/sycl/Makefile
@@ -27,13 +27,16 @@ ifeq ($(HIP),yes)
     CFLAGS += -fsycl-targets=amdgcn-amd-amdhsa -Xsycl-target-backend --offload-arch=$(HIP_ARCH)
 endif
 
-TARGETS = adamw attention_backward attention_forward classifier_fused crossentropy_forward crossentropy_softmax_backward encoder_backward encoder_forward gelu_backward gelu_forward global_norm layernorm_backward layernorm_forward matmul_backward matmul_backward_bias matmul_forward residual_forward  trimat_forward fused_residual_forward
+TARGETS = $(patsubst %.cpp,%,$(wildcard *.cpp))
 
 all: $(TARGETS)
 
 matmul_forward: matmul_forward.cpp
 	$(CC) $(CFLAGS) $< -ldnnl -o $@
 
+softmax_forward: softmax_forward.cpp
+	$(CC) $(CFLAGS) $< -fp-model=precise -o $@
+
 trimat_forward: trimat_forward.cpp
 	$(CC) $(CFLAGS) $< -ldnnl -o $@