Skip to content

Commit f5b5f5c

Browse files
committed
conv2d direct always on for opencl
1 parent b06ddf9 commit f5b5f5c

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ if (SD_OPENCL)
5858
message("-- Use OpenCL as backend stable-diffusion")
5959
set(GGML_OPENCL ON)
6060
add_definitions(-DSD_USE_OPENCL)
61+
add_definitions(-DSD_USE_CONV2D_DIRECT)
6162
endif ()
6263

6364
if (SD_HIPBLAS)

ggml_extend.hpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1510,9 +1510,12 @@ class Conv2d : public UnaryBlock {
15101510
if (bias) {
15111511
b = params["bias"];
15121512
}
1513+
#if defined(SD_USE_OPENCL)
1514+
direct = true
1515+
#endif
15131516
if (direct) {
15141517
#if defined(SD_USE_CONV2D_DIRECT)
1515-
#if defined(SD_USE_CUDA) || defined(SD_USE_SYCL) || defined(SD_USE_METAL) || defined(SD_USE_OPENCL)
1518+
#if defined(SD_USE_CUDA) || defined(SD_USE_SYCL) || defined(SD_USE_METAL)
15161519
return ggml_nn_conv_2d(ctx, x, w, b, stride.second, stride.first, padding.second, padding.first, dilation.second, dilation.first);
15171520
#else
15181521
return ggml_nn_conv_2d_direct(ctx, x, w, b, stride.second, stride.first, padding.second, padding.first, dilation.second, dilation.first);

0 commit comments

Comments
 (0)