#include "tile_program.hpp" #include "ck/tensor_operation/gpu/device/tensor_layout.hpp" #include "ck/tensor_operation/gpu/device/device_batched_gemm_gemm.hpp" #include "ck/tensor_operation/gpu/device/gemm_specialization.hpp" #include "ck/tensor_operation/gpu/device/matrix_padder.hpp" #include "ck/tensor_operation/gpu/grid/gridwise_batched_gemm_gemm_xdl_cshuffle_v1.hpp" #include "ck/host_utility/device_prop.hpp" #include "ck/host_utility/kernel_launch.hpp" #include "ck/host_utility/io.hpp" #include "ck/library/utility/device_memory.hpp" // program struct HelloWorld { __host__ __device__ void operator()(TileProgram& tp, int x, int y) { auto desc = tp.make_naive_tensor_descriptor_packed(ck::make_tuple(x)); printf("length %d\n", desc.GetLength(ck::Number<0>{})); } }; int main() { int x = 100; int y = 101; launch(HelloWorld{}, 1, 1, x, y); return 0; }