Skip to content

Commit d46e284

Browse files
author
ssjia
committed
[ET-VK][testing] Add Skycastle GPU host test workflow via ECOD
Pull Request resolved: #18170 Create a new Skycastle workflow that mirrors the existing eureka Android device test workflow but runs on an ECOD NEUTRON_NVIDIA_A10M Linux VM with an NVIDIA GPU. This enables testing the Vulkan backend on real NVIDIA hardware in CI. New files: - gpu_host_utils.sky: Utility functions for ECOD GPU host interaction (reserve, upload, ssh, release), analogous to riot_utils.sky but using ecod CLI instead of adb. - executorch_vulkan_gpu_unit_tests.sky: Main workflow (etvk-gpu-host-tests) that exports models, reserves an ECOD GPU VM, runs all tests (gtest binaries, llama runner, classification, greenscreen, scenex, skin seg), and releases the VM. Modified files: - executorch_vulkan_test_utils.sky: Added build_target_for_host() to build test binaries for x86_64 Linux host (no Android cross-compilation). ghstack-source-id: 352218807 @exported-using-ghexport Differential Revision: [D96233257](https://our.internmc.facebook.com/intern/diff/D96233257/)
1 parent 89b938a commit d46e284

File tree

2 files changed

+0
-11
lines changed

2 files changed

+0
-11
lines changed

backends/vulkan/test/custom_ops/choose_qparams_per_row.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -255,10 +255,6 @@ void choose_qparams_per_channel_reference_impl(TestCase& test_case) {
255255
const ValueSpec& input_spec = test_case.inputs()[idx++];
256256
const ValueSpec& quant_min_spec = test_case.inputs()[idx++];
257257
const ValueSpec& quant_max_spec = test_case.inputs()[idx++];
258-
const ValueSpec& eps_spec = test_case.inputs()[idx++];
259-
const ValueSpec& dtype_spec = test_case.inputs()[idx++];
260-
(void)eps_spec; // Unused in reference implementation
261-
(void)dtype_spec; // Unused in reference implementation
262258

263259
// Extract output specifications
264260
ValueSpec& scale_out_spec = test_case.outputs()[0];

backends/vulkan/test/custom_ops/q4gsw_linear.cpp

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -499,13 +499,6 @@ void reference_impl(TestCase& test_case) {
499499
}
500500

501501
int64_t quantized_linear_flop_calculator(const TestCase& test_case) {
502-
int input_idx = 0;
503-
int weight_idx = 1;
504-
if (test_case.operator_name().find("dq8ca") != std::string::npos) {
505-
input_idx = 0;
506-
weight_idx = 3; // Weight comes after input, input_scale, input_zero_point
507-
}
508-
509502
// Get input and weight dimensions
510503
const auto& input_sizes = test_case.inputs()[0].get_tensor_sizes();
511504
const auto& output_sizes = test_case.outputs()[0].get_tensor_sizes();

0 commit comments

Comments
 (0)