cleanup

Signed-off-by: Anatoly Myachev <[email protected]>
intel · Sep 19, 2024 · 49fc392 · 49fc392
1 parent bbd64f3
commit 49fc392
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py b/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py
@@ -225,7 +225,6 @@ def benchmark(Z, H, N_CTX, D_HEAD, provider):
 
     elif provider == 'triton':
         triton_fn = lambda: forward(q, k, v, causal, sm_scale)
-        # FIXME: use torch sdpa for result check after https://github.com/intel/intel-xpu-backend-for-triton/issues/2042 fixed
         torch_fn = lambda: torch.nn.functional.scaled_dot_product_attention(
             q, k, v, attn_mask=None, dropout_p=0.0, is_causal=False, scale=sm_scale).to(torch.float32)
         atol = 1e-1 if N_CTX == 16384 else 1e-2

diff --git a/third_party/intel/backend/driver.py b/third_party/intel/backend/driver.py
@@ -477,7 +477,6 @@ def get_current_target(self):
         device = self.get_current_device()
         dev_property = torch.xpu.get_device_capability(device)
         warp_size = 32
-        print(f"DEV_PROPERTY: {dev_property}")
         return GPUTarget("xpu", dev_property, warp_size)
 
     @staticmethod