indigo/LTX-2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Automated PR - 2026-07-07

Browse Source
This commit is contained in:
github-actions[bot]
2026-07-07 16:57:50 +00:00
parent 780984275f
commit 63fd9a4f86
157 changed files with 15976 additions and 5043 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/ltx-kernels/csrc/blockwise/static_switch.hpp
+60
View File
@@ -0,0 +1,60 @@
#pragma once
#define DIM_SWITCH(VAR_NAME, CONST_NAME, ...) \
if (VAR_NAME == 4096) { \
constexpr static int CONST_NAME = 4096; \
__VA_ARGS__ \
} else if (VAR_NAME == 2048){ \
constexpr static int CONST_NAME = 2048; \
__VA_ARGS__ \
} else if (VAR_NAME == 8192){ \
constexpr static int CONST_NAME = 8192; \
__VA_ARGS__ \
} else if(VAR_NAME == 16384) { \
constexpr static int CONST_NAME = 16384; \
__VA_ARGS__ \
} else { \
TORCH_CHECK(false, "Unsupported DIM_SWITCH value: ", VAR_NAME); \
}
#define BOOL_SWITCH(COND, CONST_NAME, ...) \
if (COND) { \
constexpr static bool CONST_NAME = true; \
__VA_ARGS__ \
} else { \
constexpr static bool CONST_NAME = false; \
__VA_ARGS__ \
} \
//K/128
#define BLOCK_K_SWITCH(COSNT_NAME, ...) \
if (K == 2048) { \
constexpr static int COSNT_NAME = 16; \
__VA_ARGS__ \
} \
else if (K == 4096) { \
constexpr static int COSNT_NAME = 32; \
__VA_ARGS__ \
} else if (K == 8192) { \
constexpr static int COSNT_NAME = 64; \
__VA_ARGS__ \
} else if (K == 16384) { \
constexpr static int COSNT_NAME = 128; \
__VA_ARGS__ \
} else { \
TORCH_CHECK(false, "Unsupported K value: ", K); \
}
#define M_SWITCH(...) \
if (M <= 1024) { \
constexpr static int BM = 128; \
constexpr static int BN = 128; \
constexpr static int WARP_ROW = 2; \
constexpr static int WARP_COL = 2; \
__VA_ARGS__ \
} else { \
constexpr static int BM = 128; \
constexpr static int BN = 256; \
constexpr static int WARP_ROW = 2; \
constexpr static int WARP_COL = 4; \
__VA_ARGS__ \
}